Merge pull request #7 from wiserxin/main

del the repeated torch.nn.DataParallel
shibing624 · Oct 26, 2022 · c989577 · c989577
2 parents d2f7222 + 4ac8e96
commit c989577
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/textgen/t5/t5_model.py b/textgen/t5/t5_model.py
@@ -961,9 +961,6 @@ def evaluate(self, eval_dataset, output_dir, verbose=True, silent=False, **kwarg
         nb_eval_steps = 0
         model.eval()
 
-        if args.n_gpu > 1:
-            model = torch.nn.DataParallel(model)
-
         if self.args.fp16:
             from torch.cuda import amp