SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix bug train_batch_size not an int.

Browse Source 
Division makes args.train_batch_size becoming a float.
cc @thomwolf
This commit is contained in:
VictorSanh
2018-11-04 17:19:40 -05:00
parent d55c3ae83f
commit 649e9774cd
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
run_classifier.py
View File

@@ -426,7 +426,7 @@ def main():
raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format( raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format(
args.accumulate_gradients)) 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) random.seed(args.seed)
np.random.seed(args.seed) np.random.seed(args.seed)

2
run_squad.py
View File

@@ -756,7 +756,7 @@ def main():
raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format( raise ValueError("Invalid accumulate_gradients parameter: {}, should be >= 1".format(
args.accumulate_gradients)) 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) random.seed(args.seed)
np.random.seed(args.seed) np.random.seed(args.seed)