Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

clean up dead code

This commit is contained in:
William Falcon 2019-07-03 17:05:20 -04:00
parent 3ed02e4ed6
commit 522af58504
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_lightning/callbacks/pt_callbacks.py
View File

@ -1,6 +1,6 @@
import numpy as np
import os, shutil
from pytorch_lightning.pt_overrides.override_data_parallel import LightningDataParallel
from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel
class Callback(object):
@ -34,7 +34,7 @@ class Callback(object):
self.params = params
def set_model(self, model):
if type(model) is LightningDataParallel:
if type(model) is LightningDistributedDataParallel:
model = model.module
self.model = model