classy
classy copied to clipboard
set the exact number of epochs a model should train
Is your feature request related to a problem? Please describe. There is no way for a user to explicit the number of epochs the model should train.
Describe the solution you'd like We can use the max_epochs and min_epochs parameters of the pytorch-lightning trainer to let the user set either the exact or the max/min number of epochs.