Skip to content

Draft: (Old) Migration to pytorch lightning

André Anjos requested to merge move-to-lightning into main

Started implementing pytorch ligthing. Modified requirements to include lightning, implemented LittleWNet with lightning, implemented callbacks and accelerator handling, modified ressource monitor and checkpointers following work of @dcarron on ptbench and modified training engine and script.

Merge request reports

Loading