1 files changed, 1 insertions, 1 deletions
diff --git a/train.py b/train.py
index 4a68f41..adbee8f 100644
--- a/train.py
+++ b/train.py
@@ -227,7 +227,7 @@ def main(exp, frame_sizes, dataset, **params):
             'validation_loss',
             'test_loss',
             'time'
-        ])
+        ], [(100, 'iteration'), (1, 'epoch')])
     )
     trainer.register_plugin(StatsPlugin(
         results_path,