fix errors in ensemble output saver of models trained on train datase…

…t only
vlgiitr · Nov 6, 2018 · e8a806c · e8a806c
1 parent c80ae99
commit e8a806c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Ensemble_Models/Model_OutputSaver_TrainDataset.py b/Ensemble_Models/Model_OutputSaver_TrainDataset.py
@@ -364,7 +364,7 @@ def forward(self, x, y):
 # Pretrained EmotiW DenseNet (DenseNet161_EmotiW)
 #---------------------------------------------------------------------------
 
-global_model = torch.load('../TrainedModels/TrainDataset/DenseNet161_EmotiW', map_location=lambda storage, loc: storage).module
+global_model = torch.load('../TrainedModels/TrainDataset/DenseNet161_EmotiW', map_location=lambda storage, loc: storage)
 model1 = global_model
 print('Pretrained EmotiW DenseNet Loaded! (Model 1)')
 

diff --git a/Ensemble_Models/Model_Test_OutputSaver_TrainDataset.py b/Ensemble_Models/Model_Test_OutputSaver_TrainDataset.py
@@ -313,7 +313,7 @@ def forward(self, x, y):
 # Pretrained EmotiW DenseNet (DenseNet161_EmotiW)
 #---------------------------------------------------------------------------
 
-global_model = torch.load('../TrainedModels/TrainDataset/DenseNet161_EmotiW', map_location=lambda storage, loc: storage).module
+global_model = torch.load('../TrainedModels/TrainDataset/DenseNet161_EmotiW', map_location=lambda storage, loc: storage)
 model1 = global_model
 print('Pretrained EmotiW DenseNet Loaded! (Model 1)')