Skip to content
Permalink
Browse files

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

…t only
  • Loading branch information...
dakshitagrawal97 authored and aarushgupta committed Oct 3, 2018
1 parent c80ae99 commit e8a806c5609f1d18afec669ba9eb86ddf57af22b
@@ -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)')

@@ -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)')

0 comments on commit e8a806c

Please sign in to comment.
You can’t perform that action at this time.