update eval.py

cliang1453 · May 27, 2021 · 66753c9 · 66753c9
1 parent bcf4f4d
commit 66753c9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/eval.py b/eval.py
@@ -34,8 +34,9 @@ def evaluate(args, model, tokenizer, labels, pad_token_label_id, best, mode, pre
     eval_dataloader = DataLoader(eval_dataset, sampler=eval_sampler, batch_size=args.eval_batch_size)
 
     # multi-gpu evaluate
-    if args.n_gpu > 1:
-        model = torch.nn.DataParallel(model)
+    #if args.n_gpu > 1:
+    #    model = torch.nn.DataParallel(model)
+    #model.to(args.device)
 
     logger.info("***** Running evaluation %s *****", prefix)
     if verbose: