AllenCell / AllenCell/aics-ml-segmentation
refactor with pytorch-lightning
Open
enhancement
- Dominant language
- Python
- Stars
- 32
- Forks
- 11
- PR merge metrics
- No merged PRs in 30d
Description
adopt pytorch lightning as the foundation for (1) easy multiple-GPU training/inference, (2) standard boilerplate, (3) more funality in pytorch lighnting making training more efficent (e.g., stochastic weight averaging, automatic learning rate finder, adaptive batch size, etc.)
Contributor guide
Assessment
This issue has not been assessed yet.