diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 68be6fcf..3d242ada 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -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