Moved code to lightning

Updating codebase to make use of pytorch-lightning

• Main train/valid loops are handled by lightning
• Removed device-handling code
• Each model implements its own train/valid/predict steps
• Updated config files for use with lightning
• Using lightning checkpointing and logging callbacks
• Updated DeviceMonitor to be compatible with lightning

added 1 commit

• 51efda26 - Removed checkpointer tests as we are using the lightning callback

Compare with previous version

added 1 commit

• c60bd4f0 - Updated tests

Compare with previous version

added 1 commit

• f53b7d95 - Cast device number to int

Compare with previous version

added 1 commit

• 2a7513e8 - Assign CUDA_VISIBLE_DEVICES if device specified

Compare with previous version

added 1 commit

• d711a4fa - Fixed accelerator assignement

Compare with previous version

added 2 commits

• 99356e82 - Fix saving and loading model hyperparameters
• c41430c7 - Updated model paths in tests

Compare with previous version

added 1 commit

• 4964077c - Updated submodule

Compare with previous version

added 1 commit

• 55f4eac1 - Convert labels to float instead of double

Compare with previous version

added 1 commit

• 0a858361 - Updated accelerator selection during prediction

Compare with previous version

added 1 commit

• 27647fcc - Fixed typo in docstring

Compare with previous version

added 1 commit

• 06ff7884 - Added pytorch-lightning to auto intersphinx

Compare with previous version

added 3 commits

• 7e55b4be - Replaced imports of pytorch_lightning py lightning.pytorch
• 0dfd3585 - Manually fixed path to correct lightning documentation
• 0baca28d - Updated extras.inv to fix issues with lightning's doc

Compare with previous version

added 1 commit

• b3b1462b - Changed pytorch-lightning requirement to lightning

Compare with previous version

added 1 commit

• f3cc3c55 - Updated docstrings

Compare with previous version

requested review from @ojimenez

added 3 commits

• 2bc861d4 - Removed criterion assignment to model in config
• e26fe458 - Removed ununsed colors list
• 507a6207 - Move criterion to selected device

Compare with previous version

changed title from Draft: Moved prediction to lightning to Draft: Moved code to lightning

mentioned in merge request !5 (merged)

added 37 commits

• 507a6207...a6ff407d - 27 earlier commits
• 37b723ca - completed documentation for tbx11k
• 78010d30 - added complete tbx11k dataset to project .toml
• 727dfa94 - added tests for all tbx11k configs
• 47647a7c - fixed formatting in tests
• 5f644c0a - fixed wrong paranthesis
• 9b297231 - removed unusable tbx11k custom split 3
• 2271683d - fixed small bugs in documentation
• f9a3e2d0 - more doc bug fixes
• 2577fb3c - more doc bug fixes
• f87e433e - Merge branch 'tbx11k' into 'move-to-lightning'

Compare with previous version

approved this merge request