ModelSaver(num_of_epochs: int, iterations: List, filename: str)¶
Callback to save model during training (when specified).
__init__(num_of_epochs: int, iterations: List, filename: str)¶
num_of_epochs – Total number of training epochs.
iterations – Number of total iterations or list of iterations at which to save the model.
filename – Filename to save model.
Allow saving model during training when specified in iterations.