Test Time model loading error in google colab #14
Description
trained model: models/m-25012018-123527.pth.tar
in ChexnetTrainer_test(pathDirData, pathFileTest, pathModel, nnArchitecture, nnClassCount, nnIsTrained, trBatchSize, transResize, transCrop, launchTimeStamp)
16
17 modelCheckpoint = torch.load(pathModel)
---> 18 model.load_state_dict(modelCheckpoint['state_dict'])
19
20 #-------------------- SETTINGS: DATA TRANSFORMS, TEN CROPS
................
RuntimeError: Error(s) in loading state_dict for DataParallel:
Missing key(s) in state_dict: "module.densenet121.features.denseblock1.denselayer1.norm1.weight", "module.densenet121.features.denseblock1.denselayer1.norm1.bias", "module.densenet121.features.denseblock1.denselayer1.norm1.running_mean", "module.densenet121.features.denseblock1.denselayer1.norm1.running_var", "module.densenet121.features.denseblock1.denselayer1.conv1.weight", "module.densenet121.features.denseblock1.denselayer1.norm2.weight", "module.densenet121.features.denseblock1.denselayer1.norm2.bias", "module.densenet121.features.denseblock1.denselayer1.norm2.running_mean", "module.densenet121.features.denseblock1.denselayer1.norm2.running_var", "module.densenet121.features.denseblock1.denselayer1.conv2.weight", "module.densenet121.features.denseblock1.denselayer2.norm1.weight", "module.densenet121.features.denseblock1.denselayer2.norm1.bias", "module.densenet121.features.denseblock1.denselayer2.norm1.running_mean", "module.densenet121.features.denseblock1.denselayer2.norm1.running_var", "module.densenet121.features.denseblock1.denselayer2.conv1.weight", "module.densenet121.features.denseblock1.denselayer2.norm2.weight", "module.densenet121.features.denseblock1.denselayer2.norm2.bias", "module.densenet121.features.denseblock1.denselayer2.norm2.running_mean", "module.densenet121.features.denseblock1.denselayer2.norm2.running_var", "module.densenet121.features.denseblock1.denselayer2.conv2.weight", "module.densenet121.features.denseblock1.denselayer3.norm1.weight", "m...
Unexpected key(s) in state_dict: "module.densenet121.features.denseblock1.denselayer1.norm.1.weight", "module.densenet121.features.denseblock1.denselayer1.norm.1.bias", "module.densenet121.features.denseblock1.denselayer1.norm.1.running_mean", "module.densenet121.features.denseblock1.denselayer1.norm.1.running_var", "module.densenet121.features.denseblock1.denselayer1.conv.1.weight", "module.densenet121.features.denseblock1.denselayer1.norm.2.weight", "module.densenet121.features.denseblock1.denselayer1.norm.2.bias", "module.densenet121.features.denseblock1.denselayer1.norm.2.running_mean", "module.densenet121.features.denseblock1.denselayer1.norm.2.running_var", "module.densenet121.features.denseblock1.denselayer1.conv.2.weight", "module.densenet121.features.denseblock1.denselayer2.norm.1.weight", "module.densenet121.features.denseblock1.denselayer2.norm.1.bias", "module.densenet121.features.denseblock1.denselayer2.norm.1.running_mean", "module.densenet121.features.denseblock1.denselayer2.norm.1.running_var", "module.densenet121.features.denseblock1.denselayer2.conv.1.weight", "module.densenet121.features.denseblock1.denselayer2.norm.2.weight", "module.densenet121.features.denseblock1.denselayer2.norm.2.bias", "module.densenet121.features.denseblock1.denselayer2.norm.2.running_mean", "module.densenet121.features.denseblock1.denselayer2.norm.2.running_var", "module.densenet121.features.denseblock1.denselayer2.conv.2.weight", "module.densenet121.features.denseblock1.densel...