stoic21 use lightning instead of manager

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)

Apache License 2.0

137 stars 34 forks source link

Closed mosheraboh closed 2 years ago

mosheraboh commented 2 years ago

runner_stoic21.py is the main file to review when moving to use PyTorch lightning-based trainer. All the rest are minors.