Fix run squad evaluate during training

2019-12-16 16:31:38 -05:00
parent 855ff0e91d
commit d803409215
1 changed files with 1 additions and 1 deletions
--- a/examples/run_squad.py
+++ b/examples/run_squad.py
@@ -223,7 +223,7 @@ def evaluate(args, model, tokenizer, prefix=""):
    eval_dataloader = DataLoader(dataset, sampler=eval_sampler, batch_size=args.eval_batch_size)

    # multi-gpu evaluate
-    if args.n_gpu > 1:
+    if args.n_gpu > 1 and not isinstance(model, torch.nn.DataParallel):
        model = torch.nn.DataParallel(model)

    # Eval!