add

app.py

@@ -118,7 +118,7 @@ class BaseTrainer(object):
         logger.info(f"Init test dataloader success")
         model_module = __import__(f"models.{args.model}", fromlist=["something"])
         
-        self.model = torch.nn.DataParallel(getattr(model_module, args.g_name)(args), args.gpus).cuda()
+        self.model = getattr(model_module, args.g_name)(args).cuda()
         
         if self.rank == 0:
             logger.info(self.model)