50c88091cbb82591db49f5328de8069d6d464bc7,base/base_trainer.py,BaseTrainer,__init__,#BaseTrainer#,15

Before Change

        if config["cuda"] and not torch.cuda.is_available():
            self.logger.warning("Warning: There\"s no GPU available on this machine, "
                                "training will be performed on CPU.")
        self.device = torch.device("cuda:" + str(config["gpu"]) if self.with_cuda else "cpu")
        self.model = model.to(self.device)

        self.loss = loss

After Change

            n_gpu_use = n_gpu
        self.device = torch.device("cuda:0" if n_gpu_use > 0 else "cpu")
        self.model = model.to(self.device)
        if n_gpu_use > 1:
            self.model = torch.nn.DataParallel(model, device_ids=list(range(n_gpu_use)))

        self.loss = loss
        self.metrics = metrics
        self.optimizer = optimizer

Instances