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

auto state-dict and remove the way the model is loaded during hpc

Browse Source
This commit is contained in:
William Falcon
2019-07-26 21:38:06 -04:00
parent e2c7fa44b7
commit aacf1947ea
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/model_saving.py
+1
View File
@@ -4,6 +4,7 @@ import re
import pdb
from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel, LightningDataParallel
class ModelIO(object):
def load_model_specific(self, checkpoint):