Pytorch support - #276

Draft

cpozogonzalez wants to merge 121 commits into

mainfrom

pytorch_support

Commits on Jul 21, 2026

add pytorch template

tjarkmiener
authored and
cpozogonzalez
committed
First commit with new structure to support MultiDLFrameworks

pguzman
authored and
cpozogonzalez
committed
some bug fixed and refactored some part

pguzman
authored and
cpozogonzalez
committed
refactored to avoid dependencies with tensorflow and pytorch. Added tqdm in keras.

pguzman
authored and
cpozogonzalez
committed
fixed a missing var when using pytorch and removed commented code.

pguzman
authored and
cpozogonzalez
committed
Added DataLoader in Factory format

pguzman
authored and
cpozogonzalez
committed
Fix unit test??

pguzman
authored and
cpozogonzalez
committed
fixed some bugs and fixed framework parameter

pguzman
authored and
cpozogonzalez
committed
Added nets, load checkpoints in pytorch

pguzman
authored and
cpozogonzalez
committed
fixed unit test

pguzman
authored and
cpozogonzalez
committed
formated code (round 1) and fixed a function that does not exist in pytorch legacy code

pguzman
authored and
cpozogonzalez
committed
fixed the training command binary

pguzman
authored and
cpozogonzalez
committed
Fixed flags in training and fixed a bug in validation index

pguzman
authored and
cpozogonzalez
committed
Fixed __main__ and minor changes

pguzman
authored and
cpozogonzalez
committed
refactored keras files and fixed minor issues...

pguzman
authored and
cpozogonzalez
committed
added first sketch of prediction

pguzman
authored and
cpozogonzalez
committed
fixed _predict_with_model

pguzman
authored and
cpozogonzalez
committed
updated pytorch_loader

pguzman
authored and
cpozogonzalez
committed
skeleton for training in pytorch

pguzman
authored and
cpozogonzalez
committed
fixed cuda memory problem in pytorch

pguzman
authored and
cpozogonzalez
committed
fixed parameter log saving and added missing files

pguzman
authored and
cpozogonzalez
committed
added some comments

pguzman
authored and
cpozogonzalez
committed
fixed some bugs in training and dataloader

pguzman
authored and
cpozogonzalez
committed
fixing training direction

pguzman
authored and
cpozogonzalez
committed
cleaning up pytorch training module.

pguzman
authored and
cpozogonzalez
committed
added the fix for CI

pguzman
authored and
cpozogonzalez
committed
removed unused code

pguzman
authored and
cpozogonzalez
committed
fixed minor bugs

pguzman
authored and
cpozogonzalez
committed
removed unused call

pguzman
authored and
cpozogonzalez
committed
fixed some bugs in task type. Still there is a bug to fix

pguzman
authored and
cpozogonzalez
committed
fixed the bug in Type training (blocking)

pguzman
authored and
cpozogonzalez
committed
Added Augmentation and removed unused code.

pguzman
authored and
cpozogonzalez
committed
Added Diffusion model and adapted the type loss

pguzman
authored and
cpozogonzalez
committed
Fixed Diffusion Regression code and plot bug fix

pguzman
authored and
cpozogonzalez
committed
clean up and added DBB in diffusion model

pguzman
authored and
cpozogonzalez
committed
fixed some minor bugs

pguzman
authored and
cpozogonzalez
committed
rem,oved commented code and added gpu memory trick

pguzman
authored and
cpozogonzalez
committed
modification in training and added another network for testing

pguzman
authored and
cpozogonzalez
committed
fixed bug in pytorch loader

pguzman
authored and
cpozogonzalez
committed
Adds new models
cpozogonzalez
committed
Update the git ignore
cpozogonzalez
committed
Change the pyproyect file, to the new structure of folders
cpozogonzalez
committed
Add the Subarray Descriptions and refactoring the predict task using keras
cpozogonzalez
committed
Separete the model and update the references
cpozogonzalez
committed
Add some changes in the predictions tasks
cpozogonzalez
committed
Fix the problems related with the predict model and add the loading bar
cpozogonzalez
committed
Add the MC dl2 predictions
cpozogonzalez
committed
Minor changes and optimization
cpozogonzalez
committed
Support the last dl1-data-handler version
cpozogonzalez
committed
Fix the pytorch loader and add some changes in the Pylithing validation
cpozogonzalez
committed
Refactoring CTLearnPL
cpozogonzalez
committed
Update the predictions tools and the ThinResNet model DBB
cpozogonzalez
committed
Add the warning message about the imbalance classes.
cpozogonzalez
committed
Add the class weight informations
cpozogonzalez
committed
Remove the weight calculation from the energy and cameradirrection tasks
cpozogonzalez
committed
Add the warnings for class weights
cpozogonzalez
committed
Add the weight only if the task is type, inside the CTLearnPL.py
cpozogonzalez
committed
Update the pytorch dataloader unit test
cpozogonzalez
committed
Fix the problems with preditions
cpozogonzalez
committed
Update the feuture vectors storage
cpozogonzalez
committed
Remove the comment in the imports
cpozogonzalez
committed
Apply some changes to solve the problems after rebase
cpozogonzalez
committed
Update the test CI, to be more conservative with the space, and update the pytorch loader.
cpozogonzalez
committed
Separete, the env_files in to differents (pytorch and tensorflow) and optimazate the CI
cpozogonzalez
committed
Update the name in the env_file and change the CI
cpozogonzalez
committed
Fix the predict problems after the rebase
cpozogonzalez
committed
Optimaze batch size
cpozogonzalez
committed
Update the enviroments files/*.env
cpozogonzalez
committed
Add some comments in all the functions and class and the multiGPU training support
cpozogonzalez
committed
Add the log scaling
cpozogonzalez
committed
Unify the keras and pytorch pipelines with the CTApipe Traitlets
cpozogonzalez
committed
Update the unit test and fix the onnx train problems
cpozogonzalez
committed
Update the env files
cpozogonzalez
committed
Change the name of the envioroments
cpozogonzalez
committed
Solve the problem in the FLoat loading
cpozogonzalez
committed
Add the numbers of workers etc options for the dataloader.
cpozogonzalez
committed
Add a warning in the pytroch onxx train
cpozogonzalez
committed
Add the Pytorch Keras Resnet Model
cpozogonzalez
committed
Update the enviroment files
cpozogonzalez
committed
Change the energy criterion by default to mean
cpozogonzalez
committed
Solve some bug after the rebase
cpozogonzalez
committed

Commits on Jul 22, 2026

Commits on Jul 23, 2026

Commits on Jul 28, 2026

Commits on Jul 29, 2026