set dp as default backend

2026-09-11 12:31:23 +08:00 · 2019-07-18 11:49:28 -04:00
parent d49a83dec0
commit 22f4d6e26e
1 changed files with 1 additions and 0 deletions
@@ -366,6 +366,7 @@ class Trainer(TrainerIO):
        # filter out the weights that were done on gpu so we can load on good old cpus
        self.optimizers = model.configure_optimizers()

+        model.cuda()
        model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids)

        # run through amp wrapper