removed save model logging

This commit is contained in:
William Falcon 2019-07-25 14:36:22 -04:00
parent 08bf9e16ae
commit 7e728d97e7
1 changed files with 0 additions and 1 deletions

View File

@ -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