From ea6e6ee1040fa85f743ab50b699fbeb04d9a0522 Mon Sep 17 00:00:00 2001
From: jules <jules@asdf.us>
Date: Tue, 20 Mar 2018 23:35:18 +0100
Subject: scripts

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

(limited to 'train.py')

diff --git a/train.py b/train.py
index e3061c9..b808cee 100644
--- a/train.py
+++ b/train.py
@@ -255,7 +255,7 @@ def main(exp, frame_sizes, dataset, **params):
 
     init_comet(params, trainer)
 
-    trainer.run(params['epoch_limit'])
+    trainer.run(int(params['epoch_limit']))
 
 
 if __name__ == '__main__':
-- 
cgit v1.2.3-70-g09d2