removed self.model refs

2026-09-09 11:32:07 +08:00 · 2019-06-26 18:23:02 -04:00
parent 11b4bc3fbc
commit 8f9672603b
1 changed files with 1 additions and 0 deletions
@@ -268,6 +268,7 @@ class Trainer(TrainerIO):

        # put on gpu if needed
        if self.on_gpu:
+            model.cuda()
            model = LightningDataParallel(model, device_ids=self.data_parallel_device_ids)

        # run tiny validation to make sure program won't crash during val