textgenrnn icon indicating copy to clipboard operation
textgenrnn copied to clipboard

allow save_epochs=0 to never save model weights while training

Open jkatofsky opened this issue 4 years ago • 0 comments

I have a use case where I only want to save the model weights at the end of the training. This PR adds a check for save_epochs<=0 in the on_epoch_end callback to allow disabling all weights saving while training.

EDIT: sorry for the extra commits; did a rebase to master.

jkatofsky avatar Jul 23 '20 15:07 jkatofsky