Skip to content

GitLab

Commit 649e9774 authored by VictorSanh's avatar VictorSanh
Browse files

Fix bug train_batch_size not an int. 

Division makes args.train_batch_size becoming a float.
cc @thomwolf
parent d55c3ae8
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
run_classifier.py
View file @ 649e9774
......@@ -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)
......
run_squad.py
View file @ 649e9774
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment