summaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-05-14 19:15:58 +0200
committerJules Laplace <julescarbon@gmail.com>2018-05-14 19:15:58 +0200
commit60fb2b7c87b7e6aa179c6a973a8d6e39cbe7c594 (patch)
tree8a738a43e8583f38f151cdc643a38b5a9437cda2 /train.py
parent9766ef0f3a539be7ee68bb93918f25a3298afe39 (diff)
parente2d8a6f26c5e44d970d7c069f171105376835495 (diff)
Merge branch 'master' of asdf.us:samplernn
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 1f28396..1f4ab3b 100644
--- a/train.py
+++ b/train.py
@@ -256,7 +256,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__':