wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed save model logging

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/model_saving.py
-1
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