From 7ac3311e487541b6b8a6a43a39c23ea343da3545 Mon Sep 17 00:00:00 2001 From: Baoyang Song Date: Wed, 6 Feb 2019 19:36:08 +0100 Subject: [PATCH] Fix the undefined variable in squad example --- examples/run_squad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_squad.py b/examples/run_squad.py index c80fa32e52..a19f2f0aad 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -873,7 +873,7 @@ def main(): train_examples = read_squad_examples( input_file=args.train_file, is_training=True, version_2_with_negative=args.version_2_with_negative) 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: num_train_optimization_steps = num_train_optimization_steps // torch.distributed.get_world_size()