From 649e9774cdee5c074634fe2eb37d1c1ed9f27a81 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sun, 4 Nov 2018 17:19:40 -0500 Subject: [PATCH] Fix bug train_batch_size not an int. Division makes args.train_batch_size becoming a float. cc @thomwolf --- run_classifier.py | 2 +- run_squad.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/run_classifier.py b/run_classifier.py index 58eb039d93..2c82bcd4c1 100644 --- a/run_classifier.py +++ b/run_classifier.py @@ -426,7 +426,7 @@ def main(): raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format( args.accumulate_gradients)) - args.train_batch_size = args.train_batch_size / args.accumulate_gradients + args.train_batch_size = int(args.train_batch_size / args.accumulate_gradients) random.seed(args.seed) np.random.seed(args.seed) diff --git a/run_squad.py b/run_squad.py index 868dc99a23..3a961079c9 100644 --- a/run_squad.py +++ b/run_squad.py @@ -756,7 +756,7 @@ def main(): raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format( args.accumulate_gradients)) - args.train_batch_size = args.train_batch_size / args.accumulate_gradients + args.train_batch_size = int(args.train_batch_size / args.accumulate_gradients) random.seed(args.seed) np.random.seed(args.seed)