baler
baler copied to clipboard
Torch Lightning Support for Baler
Torch Lightning proposes a compact the efficient way to train neural networks. It contained the predefined construction (Trainer, Tuner) that speeds up the code understanding and maintenance. Moreover, a lot of existing tricks for model tuning are already implemented there.
This pr can provide an alternative way of using the baler with the lightning framework.