mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08:00
removed self.model refs
This commit is contained in:
1 parent
42fe76f794
commit
301a4992f4
1 file changed
-1
@@ -268,7 +268,6 @@ 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
|
||||
|
||||
Reference in new issue
Block a user