Skip to content

GitLab

"examples/bertology.py" did not exist on "fd8a3556f08bbcfb9c4f3eadea6206751c1b1dd9"
Commit af62cc5f authored by thomwolf's avatar thomwolf
Browse files

fix run_squad example

parent eebc8abb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_squad.py
View file @ af62cc5f
...@@ -881,7 +881,7 @@ def main(): ...@@ -881,7 +881,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