From e65e8ecf0666a30dd01d383a47f729fb0ea8adea Mon Sep 17 00:00:00 2001
From: jules <jules@asdf.us>
Date: Thu, 22 Aug 2019 16:32:09 +0200
Subject: train log

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,
-- 
cgit v1.2.3-70-g09d2