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

make examples consistent, revert error in num_train_steps calculation

Browse Source
This commit is contained in:
Matej Svejda
2019-01-30 11:47:25 +01:00
parent 9c6a48c8c3
commit 5169069997
5 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/run_squad.py
View File

@@ -757,7 +757,7 @@ def main():
raise ValueError("Invalid gradient_accumulation_steps parameter: {}, should be >= 1".format(
args.gradient_accumulation_steps))
args.train_batch_size = int(args.train_batch_size / args.gradient_accumulation_steps)
args.train_batch_size = args.train_batch_size // args.gradient_accumulation_steps
random.seed(args.seed)
np.random.seed(args.seed)
@@ -788,8 +788,8 @@ def main():
if args.do_train:
train_examples = read_squad_examples(
input_file=args.train_file, is_training=True)
num_train_steps = int(
len(train_examples) / args.train_batch_size * args.num_train_epochs)
num_train_steps =
len(train_examples) // args.train_batch_size // args.gradient_accumulation_steps * args.num_train_epochs
# Prepare model
model = BertForQuestionAnswering.from_pretrained(args.bert_model,