summaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
authorjules <jules@asdf.us>2018-03-20 23:35:18 +0100
committerjules <jules@asdf.us>2018-03-20 23:35:18 +0100
commitea6e6ee1040fa85f743ab50b699fbeb04d9a0522 (patch)
treee056f13c3ef89c5b6b8713a7f80c837b333129af /train.py
parent4167442627b1414ff8fdc86528812b46168c656b (diff)
scripts
Diffstat (limited to 'train.py')
-rw-r--r--train.py2
1 files changed, 1 insertions, 1 deletions
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__':