removed save model logging

Lightning-AI · Jul 25, 2019 · 7e728d9 · 7e728d9
1 parent 08bf9e1
commit 7e728d9
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py
@@ -42,7 +42,6 @@ def on_hpc_load(self):
 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