From 7e728d97e7f335fe990d8c65ad079b4042fd4c46 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jul 2019 14:36:22 -0400 Subject: [PATCH] removed save model logging --- pytorch_lightning/root_module/model_saving.py | 1 - 1 file changed, 1 deletion(-) 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