From 39d04eb795043f12d95a406c293a2cbbe9998dd3 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 11:35:59 -0400 Subject: [PATCH] set dp as default backend --- pytorch_lightning/models/trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 655316ed..7fbcf6a1 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -371,8 +371,7 @@ class Trainer(TrainerIO): ) self.optimizers = optimizers - if self.on_gpu: - model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids) + model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids) self.__run_pretrain_routine(model)