diff --git a/plugins/interruptible.py b/plugins/interruptible.py index 13dbf6b..e011523 100644 --- a/plugins/interruptible.py +++ b/plugins/interruptible.py @@ -4,7 +4,7 @@ import shutil from plugins.plugins import BasePlugin from train import save_model -EVERY_N_EPOCHS = 0.3 # how often to save. integers >= 1 save at the end of every nth epoch. floats < 1 subdivide the epoch evenly (eg 0.33 = 3 subdivisions) +EVERY_N_EPOCHS = 1 # how often to save. integers >= 1 save at the end of every nth epoch. floats < 1 subdivide the epoch evenly (eg 0.33 = 3 subdivisions) class InterruptiblePlugin(BasePlugin):