From 74c21602fcd06a5be157da66e74f1fb82e8bd173 Mon Sep 17 00:00:00 2001 From: Damian Stewart Date: Sun, 10 Sep 2023 21:41:09 +0200 Subject: [PATCH] default every epoch not 3x per epoch --- plugins/interruptible.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):