removed save model logging
This commit is contained in:
parent
08bf9e16ae
commit
7e728d97e7
|
@ -42,7 +42,6 @@ class ModelIO(object):
|
|||
class TrainerIO(object):
|
||||
|
||||
def __get_model(self):
|
||||
print(type(self.model))
|
||||
is_dp_module = type(self.model) is LightningDistributedDataParallel or type(self.model) is LightningDataParallel
|
||||
model = self.model.module if is_dp_module else self.model
|
||||
return model
|
||||
|
|
Loading…
Reference in New Issue