Skip to content

GitLab

Unverified Commit 7ac3311e authored by Baoyang Song's avatar Baoyang Song Committed by GitHub
Browse files

Fix the undefined variable in squad example

parent 82291514
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_squad.py
View file @ 7ac3311e
...@@ -873,7 +873,7 @@ def main(): ...@@ -873,7 +873,7 @@ def main():
train_examples = read_squad_examples( train_examples = read_squad_examples(
input_file=args.train_file, is_training=True, version_2_with_negative=args.version_2_with_negative) input_file=args.train_file, is_training=True, version_2_with_negative=args.version_2_with_negative)
num_train_optimization_steps = int( num_train_optimization_steps = int(
len(train_dataset) / args.train_batch_size / args.gradient_accumulation_steps) * args.num_train_epochs len(train_examples) / args.train_batch_size / args.gradient_accumulation_steps) * args.num_train_epochs
if args.local_rank != -1: if args.local_rank != -1:
num_train_optimization_steps = num_train_optimization_steps // torch.distributed.get_world_size() num_train_optimization_steps = num_train_optimization_steps // torch.distributed.get_world_size()
......
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