RuntimeError: Error(s) in loading state_dict for FeatherNet: Missing key(s) in state_dict
guuboi opened this issue · 6 comments
model = models.FeatherNetB()
checkpoint = torch.load('checkpoints/FeatherNetB_bs32/_47_best.pth.tar', map_location='cpu')
model.load_state_dict(checkpoint['state_dict'])
RuntimeError: Error(s) in loading state_dict for FeatherNet:
Missing key(s) in state_dict: "features.0.0.weight", "features.0.1.weight", "features.0.1.bias", "features.0.1.running_mean", "features.0.1.running_var", "features.1.downsample.1.weight", "features.1.downsample.1.bias", "features.1.downsample.1.running_mean", "features.1.downsample.1.running_var", "features.1.downsample.2.weight", "features.1.conv.0.weight", "features.1.conv.1.weight", "features.1.conv.1.bias", "features.1.conv.1.running_mean", "features.1.conv.1.running_var", "features.1.conv.3.weight", "features.1.conv.4.weight", "features.1.conv.4.bias", "features.1.conv.4.running_mean", "features.1.conv.4.running_var", "features.2.fc.0.weight", "features.2.fc.0.bias", "features.2.fc.2.weight", "features.2.fc.2.bias", "features.3.downsample.1.weight", "features.3.downsample.1.bias", "features.3.downsample.1.running_mean", "features.3.downsample.1.running_var", "features.3.downsample.2.weight", "features.3.conv.0.weight", "features.3.conv.1.weight", "features.3.conv.1.bias", "features.3.conv.1.running_mean", "features.3.conv.1.running_var", "features.3.conv.3.weight", "features.3.conv.4.weight", "features.3.conv.4.bias", "features.3.conv.4.running_mean", "features.3.conv.4.running_var", "features.3.conv.6.weight", "features.3.conv.7.weight", "features.3.conv.7.bias", "features.3.conv.7.running_mean", "features.3.conv.7.running_var", "features.4.conv.0.weight", "features.4.conv.1.weight", "features.4.conv.1.bias", "features.4.conv.1.running_mean", "features.4.conv.1.running_var", "features.4.conv.3.weight", "features.4.conv.4.weight", "features.4.conv.4.bias", "features.4.conv.4.running_mean", "features.4.conv.4.running_var", "features.4.conv.6.weight", "features.4.conv.7.weight", "features.4.conv.7.bias", "features.4.conv.7.running_mean", "features.4.conv.7.running_var", "features.5.fc.0.weight", "features.5.fc.0.bias", "features.5.fc.2.weight", "features.5.fc.2.bias", "features.6.downsample.1.weight", "features.6.downsample.1.bias", "features.6.downsample.1.running_mean", "features.6.downsample.1.running_var", "features.6.downsample.2.weight", "features.6.conv.0.weight", "features.6.conv.1.weight", "features.6.conv.1.bias", "features.6.conv.1.running_mean", "features.6.conv.1.running_var", "features.6.conv.3.weight", "features.6.conv.4.weight", "features.6.conv.4.bias", "features.6.conv.4.running_mean", "features.6.conv.4.running_var", "features.6.conv.6.weight", "features.6.conv.7.weight", "features.6.conv.7.bias", "features.6.conv.7.running_mean", "features.6.conv.7.running_var", "features.7.conv.0.weight", "features.7.conv.1.weight", "features.7.conv.1.bias", "features.7.conv.1.running_mean", "features.7.conv.1.running_var", "features.7.conv.3.weight", "features.7.conv.4.weight", "features.7.conv.4.bias", "features.7.conv.4.running_mean", "features.7.conv.4.running_var", "features.7.conv.6.weight", "features.7.conv.7.weight", "features.7.conv.7.bias", "features.7.conv.7.running_mean", "features.7.conv.7.running_var", "features.8.conv.0.weight", "features.8.conv.1.weight", "features.8.conv.1.bias", "features.8.conv.1.running_mean", "features.8.conv.1.running_var", "features.8.conv.3.weight", "features.8.conv.4.weight", "features.8.conv.4.bias", "features.8.conv.4.running_mean", "features.8.conv.4.running_var", "features.8.conv.6.weight", "features.8.conv.7.weight", "features.8.conv.7.bias", "features.8.conv.7.running_mean", "features.8.conv.7.running_var", "features.9.conv.0.weight", "features.9.conv.1.weight", "features.9.conv.1.bias", "features.9.conv.1.running_mean", "features.9.conv.1.running_var", "features.9.conv.3.weight", "features.9.conv.4.weight", "features.9.conv.4.bias", "features.9.conv.4.running_mean", "features.9.conv.4.running_var", "features.9.conv.6.weight", "features.9.conv.7.weight", "features.9.conv.7.bias", "features.9.conv.7.running_mean", "features.9.conv.7.running_var", "features.10.conv.0.weight", "features.10.conv.1.weight", "features.10.conv.1.bias", "features.10.conv.1.running_mean", "features.10.conv.1.running_var", "features.10.conv.3.weight", "features.10.conv.4.weight", "features.10.conv.4.bias", "features.10.conv.4.running_mean", "features.10.conv.4.running_var", "features.10.conv.6.weight", "features.10.conv.7.weight", "features.10.conv.7.bias", "features.10.conv.7.running_mean", "features.10.conv.7.running_var", "features.11.conv.0.weight", "features.11.conv.1.weight", "features.11.conv.1.bias", "features.11.conv.1.running_mean", "features.11.conv.1.running_var", "features.11.conv.3.weight", "features.11.conv.4.weight", "features.11.conv.4.bias", "features.11.conv.4.running_mean", "features.11.conv.4.running_var", "features.11.conv.6.weight", "features.11.conv.7.weight", "features.11.conv.7.bias", "features.11.conv.7.running_mean", "features.11.conv.7.running_var", "features.12.fc.0.weight", "features.12.fc.0.bias", "features.12.fc.2.weight", "features.12.fc.2.bias", "features.13.downsample.1.weight", "features.13.downsample.1.bias", "features.13.downsample.1.running_mean", "features.13.downsample.1.running_var", "features.13.downsample.2.weight", "features.13.conv.0.weight", "features.13.conv.1.weight", "features.13.conv.1.bias", "features.13.conv.1.running_mean", "features.13.conv.1.running_var", "features.13.conv.3.weight", "features.13.conv.4.weight", "features.13.conv.4.bias", "features.13.conv.4.running_mean", "features.13.conv.4.running_var", "features.13.conv.6.weight", "features.13.conv.7.weight", "features.13.conv.7.bias", "features.13.conv.7.running_mean", "features.13.conv.7.running_var", "features.14.conv.0.weight", "features.14.conv.1.weight", "features.14.conv.1.bias", "features.14.conv.1.running_mean", "features.14.conv.1.running_var", "features.14.conv.3.weight", "features.14.conv.4.weight", "features.14.conv.4.bias", "features.14.conv.4.running_mean", "features.14.conv.4.running_var", "features.14.conv.6.weight", "features.14.conv.7.weight", "features.14.conv.7.bias", "features.14.conv.7.running_mean", "features.14.conv.7.running_var", "features.15.conv.0.weight", "features.15.conv.1.weight", "features.15.conv.1.bias", "features.15.conv.1.running_mean", "features.15.conv.1.running_var", "features.15.conv.3.weight", "features.15.conv.4.weight", "features.15.conv.4.bias", "features.15.conv.4.running_mean", "features.15.conv.4.running_var", "features.15.conv.6.weight", "features.15.conv.7.weight", "features.15.conv.7.bias", "features.15.conv.7.running_mean", "features.15.conv.7.running_var", "features.16.fc.0.weight", "features.16.fc.0.bias", "features.16.fc.2.weight", "features.16.fc.2.bias", "final_DW.0.weight".
Unexpected key(s) in state_dict: "module.features.0.0.weight", "module.features.0.1.weight", "module.features.0.1.bias", "module.features.0.1.running_mean", "module.features.0.1.running_var", "module.features.0.1.num_batches_tracked", "module.features.1.downsample.1.weight", "module.features.1.downsample.1.bias", "module.features.1.downsample.1.running_mean", "module.features.1.downsample.1.running_var", "module.features.1.downsample.1.num_batches_tracked", "module.features.1.downsample.2.weight", "module.features.1.conv.0.weight", "module.features.1.conv.1.weight", "module.features.1.conv.1.bias", "module.features.1.conv.1.running_mean", "module.features.1.conv.1.running_var", "module.features.1.conv.1.num_batches_tracked", "module.features.1.conv.3.weight", "module.features.1.conv.4.weight", "module.features.1.conv.4.bias", "module.features.1.conv.4.running_mean", "module.features.1.conv.4.running_var", "module.features.1.conv.4.num_batches_tracked", "module.features.2.fc.0.weight", "module.features.2.fc.0.bias", "module.features.2.fc.2.weight", "module.features.2.fc.2.bias", "module.features.3.downsample.1.weight", "module.features.3.downsample.1.bias", "module.features.3.downsample.1.running_mean", "module.features.3.downsample.1.running_var", "module.features.3.downsample.1.num_batches_tracked", "module.features.3.downsample.2.weight", "module.features.3.conv.0.weight", "module.features.3.conv.1.weight", "module.features.3.conv.1.bias", "module.features.3.conv.1.running_mean", "module.features.3.conv.1.running_var", "module.features.3.conv.1.num_batches_tracked", "module.features.3.conv.3.weight", "module.features.3.conv.4.weight", "module.features.3.conv.4.bias", "module.features.3.conv.4.running_mean", "module.features.3.conv.4.running_var", "module.features.3.conv.4.num_batches_tracked", "module.features.3.conv.6.weight", "module.features.3.conv.7.weight", "module.features.3.conv.7.bias", "module.features.3.conv.7.running_mean", "module.features.3.conv.7.running_var", "module.features.3.conv.7.num_batches_tracked", "module.features.4.conv.0.weight", "module.features.4.conv.1.weight", "module.features.4.conv.1.bias", "module.features.4.conv.1.running_mean", "module.features.4.conv.1.running_var", "module.features.4.conv.1.num_batches_tracked", "module.features.4.conv.3.weight", "module.features.4.conv.4.weight", "module.features.4.conv.4.bias", "module.features.4.conv.4.running_mean", "module.features.4.conv.4.running_var", "module.features.4.conv.4.num_batches_tracked", "module.features.4.conv.6.weight", "module.features.4.conv.7.weight", "module.features.4.conv.7.bias", "module.features.4.conv.7.running_mean", "module.features.4.conv.7.running_var", "module.features.4.conv.7.num_batches_tracked", "module.features.5.fc.0.weight", "module.features.5.fc.0.bias", "module.features.5.fc.2.weight", "module.features.5.fc.2.bias", "module.features.6.downsample.1.weight", "module.features.6.downsample.1.bias", "module.features.6.downsample.1.running_mean", "module.features.6.downsample.1.running_var", "module.features.6.downsample.1.num_batches_tracked", "module.features.6.downsample.2.weight", "module.features.6.conv.0.weight", "module.features.6.conv.1.weight", "module.features.6.conv.1.bias", "module.features.6.conv.1.running_mean", "module.features.6.conv.1.running_var", "module.features.6.conv.1.num_batches_tracked", "module.features.6.conv.3.weight", "module.features.6.conv.4.weight", "module.features.6.conv.4.bias", "module.features.6.conv.4.running_mean", "module.features.6.conv.4.running_var", "module.features.6.conv.4.num_batches_tracked", "module.features.6.conv.6.weight", "module.features.6.conv.7.weight", "module.features.6.conv.7.bias", "module.features.6.conv.7.running_mean", "module.features.6.conv.7.running_var", "module.features.6.conv.7.num_batches_tracked", "module.features.7.conv.0.weight", "module.features.7.conv.1.weight", "module.features.7.conv.1.bias", "module.features.7.conv.1.running_mean", "module.features.7.conv.1.running_var", "module.features.7.conv.1.num_batches_tracked", "module.features.7.conv.3.weight", "module.features.7.conv.4.weight", "module.features.7.conv.4.bias", "module.features.7.conv.4.running_mean", "module.features.7.conv.4.running_var", "module.features.7.conv.4.num_batches_tracked", "module.features.7.conv.6.weight", "module.features.7.conv.7.weight", "module.features.7.conv.7.bias", "module.features.7.conv.7.running_mean", "module.features.7.conv.7.running_var", "module.features.7.conv.7.num_batches_tracked", "module.features.8.conv.0.weight", "module.features.8.conv.1.weight", "module.features.8.conv.1.bias", "module.features.8.conv.1.running_mean", "module.features.8.conv.1.running_var", "module.features.8.conv.1.num_batches_tracked", "module.features.8.conv.3.weight", "module.features.8.conv.4.weight", "module.features.8.conv.4.bias", "module.features.8.conv.4.running_mean", "module.features.8.conv.4.running_var", "module.features.8.conv.4.num_batches_tracked", "module.features.8.conv.6.weight", "module.features.8.conv.7.weight", "module.features.8.conv.7.bias", "module.features.8.conv.7.running_mean", "module.features.8.conv.7.running_var", "module.features.8.conv.7.num_batches_tracked", "module.features.9.conv.0.weight", "module.features.9.conv.1.weight", "module.features.9.conv.1.bias", "module.features.9.conv.1.running_mean", "module.features.9.conv.1.running_var", "module.features.9.conv.1.num_batches_tracked", "module.features.9.conv.3.weight", "module.features.9.conv.4.weight", "module.features.9.conv.4.bias", "module.features.9.conv.4.running_mean", "module.features.9.conv.4.running_var", "module.features.9.conv.4.num_batches_tracked", "module.features.9.conv.6.weight", "module.features.9.conv.7.weight", "module.features.9.conv.7.bias", "module.features.9.conv.7.running_mean", "module.features.9.conv.7.running_var", "module.features.9.conv.7.num_batches_tracked", "module.features.10.conv.0.weight", "module.features.10.conv.1.weight", "module.features.10.conv.1.bias", "module.features.10.conv.1.running_mean", "module.features.10.conv.1.running_var", "module.features.10.conv.1.num_batches_tracked", "module.features.10.conv.3.weight", "module.features.10.conv.4.weight", "module.features.10.conv.4.bias", "module.features.10.conv.4.running_mean", "module.features.10.conv.4.running_var", "module.features.10.conv.4.num_batches_tracked", "module.features.10.conv.6.weight", "module.features.10.conv.7.weight", "module.features.10.conv.7.bias", "module.features.10.conv.7.running_mean", "module.features.10.conv.7.running_var", "module.features.10.conv.7.num_batches_tracked", "module.features.11.conv.0.weight", "module.features.11.conv.1.weight", "module.features.11.conv.1.bias", "module.features.11.conv.1.running_mean", "module.features.11.conv.1.running_var", "module.features.11.conv.1.num_batches_tracked", "module.features.11.conv.3.weight", "module.features.11.conv.4.weight", "module.features.11.conv.4.bias", "module.features.11.conv.4.running_mean", "module.features.11.conv.4.running_var", "module.features.11.conv.4.num_batches_tracked", "module.features.11.conv.6.weight", "module.features.11.conv.7.weight", "module.features.11.conv.7.bias", "module.features.11.conv.7.running_mean", "module.features.11.conv.7.running_var", "module.features.11.conv.7.num_batches_tracked", "module.features.12.fc.0.weight", "module.features.12.fc.0.bias", "module.features.12.fc.2.weight", "module.features.12.fc.2.bias", "module.features.13.downsample.1.weight", "module.features.13.downsample.1.bias", "module.features.13.downsample.1.running_mean", "module.features.13.downsample.1.running_var", "module.features.13.downsample.1.num_batches_tracked", "module.features.13.downsample.2.weight", "module.features.13.conv.0.weight", "module.features.13.conv.1.weight", "module.features.13.conv.1.bias", "module.features.13.conv.1.running_mean", "module.features.13.conv.1.running_var", "module.features.13.conv.1.num_batches_tracked", "module.features.13.conv.3.weight", "module.features.13.conv.4.weight", "module.features.13.conv.4.bias", "module.features.13.conv.4.running_mean", "module.features.13.conv.4.running_var", "module.features.13.conv.4.num_batches_tracked", "module.features.13.conv.6.weight", "module.features.13.conv.7.weight", "module.features.13.conv.7.bias", "module.features.13.conv.7.running_mean", "module.features.13.conv.7.running_var", "module.features.13.conv.7.num_batches_tracked", "module.features.14.conv.0.weight", "module.features.14.conv.1.weight", "module.features.14.conv.1.bias", "module.features.14.conv.1.running_mean", "module.features.14.conv.1.running_var", "module.features.14.conv.1.num_batches_tracked", "module.features.14.conv.3.weight", "module.features.14.conv.4.weight", "module.features.14.conv.4.bias", "module.features.14.conv.4.running_mean", "module.features.14.conv.4.running_var", "module.features.14.conv.4.num_batches_tracked", "module.features.14.conv.6.weight", "module.features.14.conv.7.weight", "module.features.14.conv.7.bias", "module.features.14.conv.7.running_mean", "module.features.14.conv.7.running_var", "module.features.14.conv.7.num_batches_tracked", "module.features.15.conv.0.weight", "module.features.15.conv.1.weight", "module.features.15.conv.1.bias", "module.features.15.conv.1.running_mean", "module.features.15.conv.1.running_var", "module.features.15.conv.1.num_batches_tracked", "module.features.15.conv.3.weight", "module.features.15.conv.4.weight", "module.features.15.conv.4.bias", "module.features.15.conv.4.running_mean", "module.features.15.conv.4.running_var", "module.features.15.conv.4.num_batches_tracked", "module.features.15.conv.6.weight", "module.features.15.conv.7.weight", "module.features.15.conv.7.bias", "module.features.15.conv.7.running_mean", "module.features.15.conv.7.running_var", "module.features.15.conv.7.num_batches_tracked", "module.features.16.fc.0.weight", "module.features.16.fc.0.bias", "module.features.16.fc.2.weight", "module.features.16.fc.2.bias", "module.final_DW.0.weight".
model = models.FeatherNetB()
checkpoint = torch.load('checkpoints/FeatherNetB_bs32/_47_best.pth.tar', map_location='cpu')
model.load_state_dict(checkpoint['state_dict'])RuntimeError: Error(s) in loading state_dict for FeatherNet:
Missing key(s) in state_dict: "features.0.0.weight", "features.0.1.weight", "features.0.1.bias", "features.0.1.running_mean", "features.0.1.running_var", "features.1.downsample.1.weight", "features.1.downsample.1.bias", "features.1.downsample.1.running_mean", "features.1.downsample.1.running_var", "features.1.downsample.2.weight", "features.1.conv.0.weight", "features.1.conv.1.weight", "features.1.conv.1.bias", "features.1.conv.1.running_mean", "features.1.conv.1.running_var", "features.1.conv.3.weight", "features.1.conv.4.weight", "features.1.conv.4.bias", "features.1.conv.4.running_mean", "features.1.conv.4.running_var", "features.2.fc.0.weight", "features.2.fc.0.bias", "features.2.fc.2.weight", "features.2.fc.2.bias", "features.3.downsample.1.weight", "features.3.downsample.1.bias", "features.3.downsample.1.running_mean", "features.3.downsample.1.running_var", "features.3.downsample.2.weight", "features.3.conv.0.weight", "features.3.conv.1.weight", "features.3.conv.1.bias", "features.3.conv.1.running_mean", "features.3.conv.1.running_var", "features.3.conv.3.weight", "features.3.conv.4.weight", "features.3.conv.4.bias", "features.3.conv.4.running_mean", "features.3.conv.4.running_var", "features.3.conv.6.weight", "features.3.conv.7.weight", "features.3.conv.7.bias", "features.3.conv.7.running_mean", "features.3.conv.7.running_var", "features.4.conv.0.weight", "features.4.conv.1.weight", "features.4.conv.1.bias", "features.4.conv.1.running_mean", "features.4.conv.1.running_var", "features.4.conv.3.weight", "features.4.conv.4.weight", "features.4.conv.4.bias", "features.4.conv.4.running_mean", "features.4.conv.4.running_var", "features.4.conv.6.weight", "features.4.conv.7.weight", "features.4.conv.7.bias", "features.4.conv.7.running_mean", "features.4.conv.7.running_var", "features.5.fc.0.weight", "features.5.fc.0.bias", "features.5.fc.2.weight", "features.5.fc.2.bias", "features.6.downsample.1.weight", "features.6.downsample.1.bias", "features.6.downsample.1.running_mean", "features.6.downsample.1.running_var", "features.6.downsample.2.weight", "features.6.conv.0.weight", "features.6.conv.1.weight", "features.6.conv.1.bias", "features.6.conv.1.running_mean", "features.6.conv.1.running_var", "features.6.conv.3.weight", "features.6.conv.4.weight", "features.6.conv.4.bias", "features.6.conv.4.running_mean", "features.6.conv.4.running_var", "features.6.conv.6.weight", "features.6.conv.7.weight", "features.6.conv.7.bias", "features.6.conv.7.running_mean", "features.6.conv.7.running_var", "features.7.conv.0.weight", "features.7.conv.1.weight", "features.7.conv.1.bias", "features.7.conv.1.running_mean", "features.7.conv.1.running_var", "features.7.conv.3.weight", "features.7.conv.4.weight", "features.7.conv.4.bias", "features.7.conv.4.running_mean", "features.7.conv.4.running_var", "features.7.conv.6.weight", "features.7.conv.7.weight", "features.7.conv.7.bias", "features.7.conv.7.running_mean", "features.7.conv.7.running_var", "features.8.conv.0.weight", "features.8.conv.1.weight", "features.8.conv.1.bias", "features.8.conv.1.running_mean", "features.8.conv.1.running_var", "features.8.conv.3.weight", "features.8.conv.4.weight", "features.8.conv.4.bias", "features.8.conv.4.running_mean", "features.8.conv.4.running_var", "features.8.conv.6.weight", "features.8.conv.7.weight", "features.8.conv.7.bias", "features.8.conv.7.running_mean", "features.8.conv.7.running_var", "features.9.conv.0.weight", "features.9.conv.1.weight", "features.9.conv.1.bias", "features.9.conv.1.running_mean", "features.9.conv.1.running_var", "features.9.conv.3.weight", "features.9.conv.4.weight", "features.9.conv.4.bias", "features.9.conv.4.running_mean", "features.9.conv.4.running_var", "features.9.conv.6.weight", "features.9.conv.7.weight", "features.9.conv.7.bias", "features.9.conv.7.running_mean", "features.9.conv.7.running_var", "features.10.conv.0.weight", "features.10.conv.1.weight", "features.10.conv.1.bias", "features.10.conv.1.running_mean", "features.10.conv.1.running_var", "features.10.conv.3.weight", "features.10.conv.4.weight", "features.10.conv.4.bias", "features.10.conv.4.running_mean", "features.10.conv.4.running_var", "features.10.conv.6.weight", "features.10.conv.7.weight", "features.10.conv.7.bias", "features.10.conv.7.running_mean", "features.10.conv.7.running_var", "features.11.conv.0.weight", "features.11.conv.1.weight", "features.11.conv.1.bias", "features.11.conv.1.running_mean", "features.11.conv.1.running_var", "features.11.conv.3.weight", "features.11.conv.4.weight", "features.11.conv.4.bias", "features.11.conv.4.running_mean", "features.11.conv.4.running_var", "features.11.conv.6.weight", "features.11.conv.7.weight", "features.11.conv.7.bias", "features.11.conv.7.running_mean", "features.11.conv.7.running_var", "features.12.fc.0.weight", "features.12.fc.0.bias", "features.12.fc.2.weight", "features.12.fc.2.bias", "features.13.downsample.1.weight", "features.13.downsample.1.bias", "features.13.downsample.1.running_mean", "features.13.downsample.1.running_var", "features.13.downsample.2.weight", "features.13.conv.0.weight", "features.13.conv.1.weight", "features.13.conv.1.bias", "features.13.conv.1.running_mean", "features.13.conv.1.running_var", "features.13.conv.3.weight", "features.13.conv.4.weight", "features.13.conv.4.bias", "features.13.conv.4.running_mean", "features.13.conv.4.running_var", "features.13.conv.6.weight", "features.13.conv.7.weight", "features.13.conv.7.bias", "features.13.conv.7.running_mean", "features.13.conv.7.running_var", "features.14.conv.0.weight", "features.14.conv.1.weight", "features.14.conv.1.bias", "features.14.conv.1.running_mean", "features.14.conv.1.running_var", "features.14.conv.3.weight", "features.14.conv.4.weight", "features.14.conv.4.bias", "features.14.conv.4.running_mean", "features.14.conv.4.running_var", "features.14.conv.6.weight", "features.14.conv.7.weight", "features.14.conv.7.bias", "features.14.conv.7.running_mean", "features.14.conv.7.running_var", "features.15.conv.0.weight", "features.15.conv.1.weight", "features.15.conv.1.bias", "features.15.conv.1.running_mean", "features.15.conv.1.running_var", "features.15.conv.3.weight", "features.15.conv.4.weight", "features.15.conv.4.bias", "features.15.conv.4.running_mean", "features.15.conv.4.running_var", "features.15.conv.6.weight", "features.15.conv.7.weight", "features.15.conv.7.bias", "features.15.conv.7.running_mean", "features.15.conv.7.running_var", "features.16.fc.0.weight", "features.16.fc.0.bias", "features.16.fc.2.weight", "features.16.fc.2.bias", "final_DW.0.weight".
Unexpected key(s) in state_dict: "module.features.0.0.weight", "module.features.0.1.weight", "module.features.0.1.bias", "module.features.0.1.running_mean", "module.features.0.1.running_var", "module.features.0.1.num_batches_tracked", "module.features.1.downsample.1.weight", "module.features.1.downsample.1.bias", "module.features.1.downsample.1.running_mean", "module.features.1.downsample.1.running_var", "module.features.1.downsample.1.num_batches_tracked", "module.features.1.downsample.2.weight", "module.features.1.conv.0.weight", "module.features.1.conv.1.weight", "module.features.1.conv.1.bias", "module.features.1.conv.1.running_mean", "module.features.1.conv.1.running_var", "module.features.1.conv.1.num_batches_tracked", "module.features.1.conv.3.weight", "module.features.1.conv.4.weight", "module.features.1.conv.4.bias", "module.features.1.conv.4.running_mean", "module.features.1.conv.4.running_var", "module.features.1.conv.4.num_batches_tracked", "module.features.2.fc.0.weight", "module.features.2.fc.0.bias", "module.features.2.fc.2.weight", "module.features.2.fc.2.bias", "module.features.3.downsample.1.weight", "module.features.3.downsample.1.bias", "module.features.3.downsample.1.running_mean", "module.features.3.downsample.1.running_var", "module.features.3.downsample.1.num_batches_tracked", "module.features.3.downsample.2.weight", "module.features.3.conv.0.weight", "module.features.3.conv.1.weight", "module.features.3.conv.1.bias", "module.features.3.conv.1.running_mean", "module.features.3.conv.1.running_var", "module.features.3.conv.1.num_batches_tracked", "module.features.3.conv.3.weight", "module.features.3.conv.4.weight", "module.features.3.conv.4.bias", "module.features.3.conv.4.running_mean", "module.features.3.conv.4.running_var", "module.features.3.conv.4.num_batches_tracked", "module.features.3.conv.6.weight", "module.features.3.conv.7.weight", "module.features.3.conv.7.bias", "module.features.3.conv.7.running_mean", "module.features.3.conv.7.running_var", "module.features.3.conv.7.num_batches_tracked", "module.features.4.conv.0.weight", "module.features.4.conv.1.weight", "module.features.4.conv.1.bias", "module.features.4.conv.1.running_mean", "module.features.4.conv.1.running_var", "module.features.4.conv.1.num_batches_tracked", "module.features.4.conv.3.weight", "module.features.4.conv.4.weight", "module.features.4.conv.4.bias", "module.features.4.conv.4.running_mean", "module.features.4.conv.4.running_var", "module.features.4.conv.4.num_batches_tracked", "module.features.4.conv.6.weight", "module.features.4.conv.7.weight", "module.features.4.conv.7.bias", "module.features.4.conv.7.running_mean", "module.features.4.conv.7.running_var", "module.features.4.conv.7.num_batches_tracked", "module.features.5.fc.0.weight", "module.features.5.fc.0.bias", "module.features.5.fc.2.weight", "module.features.5.fc.2.bias", "module.features.6.downsample.1.weight", "module.features.6.downsample.1.bias", "module.features.6.downsample.1.running_mean", "module.features.6.downsample.1.running_var", "module.features.6.downsample.1.num_batches_tracked", "module.features.6.downsample.2.weight", "module.features.6.conv.0.weight", "module.features.6.conv.1.weight", "module.features.6.conv.1.bias", "module.features.6.conv.1.running_mean", "module.features.6.conv.1.running_var", "module.features.6.conv.1.num_batches_tracked", "module.features.6.conv.3.weight", "module.features.6.conv.4.weight", "module.features.6.conv.4.bias", "module.features.6.conv.4.running_mean", "module.features.6.conv.4.running_var", "module.features.6.conv.4.num_batches_tracked", "module.features.6.conv.6.weight", "module.features.6.conv.7.weight", "module.features.6.conv.7.bias", "module.features.6.conv.7.running_mean", "module.features.6.conv.7.running_var", "module.features.6.conv.7.num_batches_tracked", "module.features.7.conv.0.weight", "module.features.7.conv.1.weight", "module.features.7.conv.1.bias", "module.features.7.conv.1.running_mean", "module.features.7.conv.1.running_var", "module.features.7.conv.1.num_batches_tracked", "module.features.7.conv.3.weight", "module.features.7.conv.4.weight", "module.features.7.conv.4.bias", "module.features.7.conv.4.running_mean", "module.features.7.conv.4.running_var", "module.features.7.conv.4.num_batches_tracked", "module.features.7.conv.6.weight", "module.features.7.conv.7.weight", "module.features.7.conv.7.bias", "module.features.7.conv.7.running_mean", "module.features.7.conv.7.running_var", "module.features.7.conv.7.num_batches_tracked", "module.features.8.conv.0.weight", "module.features.8.conv.1.weight", "module.features.8.conv.1.bias", "module.features.8.conv.1.running_mean", "module.features.8.conv.1.running_var", "module.features.8.conv.1.num_batches_tracked", "module.features.8.conv.3.weight", "module.features.8.conv.4.weight", "module.features.8.conv.4.bias", "module.features.8.conv.4.running_mean", "module.features.8.conv.4.running_var", "module.features.8.conv.4.num_batches_tracked", "module.features.8.conv.6.weight", "module.features.8.conv.7.weight", "module.features.8.conv.7.bias", "module.features.8.conv.7.running_mean", "module.features.8.conv.7.running_var", "module.features.8.conv.7.num_batches_tracked", "module.features.9.conv.0.weight", "module.features.9.conv.1.weight", "module.features.9.conv.1.bias", "module.features.9.conv.1.running_mean", "module.features.9.conv.1.running_var", "module.features.9.conv.1.num_batches_tracked", "module.features.9.conv.3.weight", "module.features.9.conv.4.weight", "module.features.9.conv.4.bias", "module.features.9.conv.4.running_mean", "module.features.9.conv.4.running_var", "module.features.9.conv.4.num_batches_tracked", "module.features.9.conv.6.weight", "module.features.9.conv.7.weight", "module.features.9.conv.7.bias", "module.features.9.conv.7.running_mean", "module.features.9.conv.7.running_var", "module.features.9.conv.7.num_batches_tracked", "module.features.10.conv.0.weight", "module.features.10.conv.1.weight", "module.features.10.conv.1.bias", "module.features.10.conv.1.running_mean", "module.features.10.conv.1.running_var", "module.features.10.conv.1.num_batches_tracked", "module.features.10.conv.3.weight", "module.features.10.conv.4.weight", "module.features.10.conv.4.bias", "module.features.10.conv.4.running_mean", "module.features.10.conv.4.running_var", "module.features.10.conv.4.num_batches_tracked", "module.features.10.conv.6.weight", "module.features.10.conv.7.weight", "module.features.10.conv.7.bias", "module.features.10.conv.7.running_mean", "module.features.10.conv.7.running_var", "module.features.10.conv.7.num_batches_tracked", "module.features.11.conv.0.weight", "module.features.11.conv.1.weight", "module.features.11.conv.1.bias", "module.features.11.conv.1.running_mean", "module.features.11.conv.1.running_var", "module.features.11.conv.1.num_batches_tracked", "module.features.11.conv.3.weight", "module.features.11.conv.4.weight", "module.features.11.conv.4.bias", "module.features.11.conv.4.running_mean", "module.features.11.conv.4.running_var", "module.features.11.conv.4.num_batches_tracked", "module.features.11.conv.6.weight", "module.features.11.conv.7.weight", "module.features.11.conv.7.bias", "module.features.11.conv.7.running_mean", "module.features.11.conv.7.running_var", "module.features.11.conv.7.num_batches_tracked", "module.features.12.fc.0.weight", "module.features.12.fc.0.bias", "module.features.12.fc.2.weight", "module.features.12.fc.2.bias", "module.features.13.downsample.1.weight", "module.features.13.downsample.1.bias", "module.features.13.downsample.1.running_mean", "module.features.13.downsample.1.running_var", "module.features.13.downsample.1.num_batches_tracked", "module.features.13.downsample.2.weight", "module.features.13.conv.0.weight", "module.features.13.conv.1.weight", "module.features.13.conv.1.bias", "module.features.13.conv.1.running_mean", "module.features.13.conv.1.running_var", "module.features.13.conv.1.num_batches_tracked", "module.features.13.conv.3.weight", "module.features.13.conv.4.weight", "module.features.13.conv.4.bias", "module.features.13.conv.4.running_mean", "module.features.13.conv.4.running_var", "module.features.13.conv.4.num_batches_tracked", "module.features.13.conv.6.weight", "module.features.13.conv.7.weight", "module.features.13.conv.7.bias", "module.features.13.conv.7.running_mean", "module.features.13.conv.7.running_var", "module.features.13.conv.7.num_batches_tracked", "module.features.14.conv.0.weight", "module.features.14.conv.1.weight", "module.features.14.conv.1.bias", "module.features.14.conv.1.running_mean", "module.features.14.conv.1.running_var", "module.features.14.conv.1.num_batches_tracked", "module.features.14.conv.3.weight", "module.features.14.conv.4.weight", "module.features.14.conv.4.bias", "module.features.14.conv.4.running_mean", "module.features.14.conv.4.running_var", "module.features.14.conv.4.num_batches_tracked", "module.features.14.conv.6.weight", "module.features.14.conv.7.weight", "module.features.14.conv.7.bias", "module.features.14.conv.7.running_mean", "module.features.14.conv.7.running_var", "module.features.14.conv.7.num_batches_tracked", "module.features.15.conv.0.weight", "module.features.15.conv.1.weight", "module.features.15.conv.1.bias", "module.features.15.conv.1.running_mean", "module.features.15.conv.1.running_var", "module.features.15.conv.1.num_batches_tracked", "module.features.15.conv.3.weight", "module.features.15.conv.4.weight", "module.features.15.conv.4.bias", "module.features.15.conv.4.running_mean", "module.features.15.conv.4.running_var", "module.features.15.conv.4.num_batches_tracked", "module.features.15.conv.6.weight", "module.features.15.conv.7.weight", "module.features.15.conv.7.bias", "module.features.15.conv.7.running_mean", "module.features.15.conv.7.running_var", "module.features.15.conv.7.num_batches_tracked", "module.features.16.fc.0.weight", "module.features.16.fc.0.bias", "module.features.16.fc.2.weight", "module.features.16.fc.2.bias", "module.final_DW.0.weight".
I met this problem too. I think maybe the definition of the model in FeatherNet.py is different from the model in the pretrain model.
model = models.FeatherNetB()
checkpoint = torch.load('checkpoints/FeatherNetB_bs32/_47_best.pth.tar', map_location='cpu')
model.load_state_dict(checkpoint['state_dict'])RuntimeError: Error(s) in loading state_dict for FeatherNet:
Missing key(s) in state_dict: "features.0.0.weight", "features.0.1.weight", "features.0.1.bias", "features.0.1.running_mean", "features.0.1.running_var", "features.1.downsample.1.weight", "features.1.downsample.1.bias", "features.1.downsample.1.running_mean", "features.1.downsample.1.running_var", "features.1.downsample.2.weight", "features.1.conv.0.weight", "features.1.conv.1.weight", "features.1.conv.1.bias", "features.1.conv.1.running_mean", "features.1.conv.1.running_var", "features.1.conv.3.weight", "features.1.conv.4.weight", "features.1.conv.4.bias", "features.1.conv.4.running_mean", "features.1.conv.4.running_var", "features.2.fc.0.weight", "features.2.fc.0.bias", "features.2.fc.2.weight", "features.2.fc.2.bias", "features.3.downsample.1.weight", "features.3.downsample.1.bias", "features.3.downsample.1.running_mean", "features.3.downsample.1.running_var", "features.3.downsample.2.weight", "features.3.conv.0.weight", "features.3.conv.1.weight", "features.3.conv.1.bias", "features.3.conv.1.running_mean", "features.3.conv.1.running_var", "features.3.conv.3.weight", "features.3.conv.4.weight", "features.3.conv.4.bias", "features.3.conv.4.running_mean", "features.3.conv.4.running_var", "features.3.conv.6.weight", "features.3.conv.7.weight", "features.3.conv.7.bias", "features.3.conv.7.running_mean", "features.3.conv.7.running_var", "features.4.conv.0.weight", "features.4.conv.1.weight", "features.4.conv.1.bias", "features.4.conv.1.running_mean", "features.4.conv.1.running_var", "features.4.conv.3.weight", "features.4.conv.4.weight", "features.4.conv.4.bias", "features.4.conv.4.running_mean", "features.4.conv.4.running_var", "features.4.conv.6.weight", "features.4.conv.7.weight", "features.4.conv.7.bias", "features.4.conv.7.running_mean", "features.4.conv.7.running_var", "features.5.fc.0.weight", "features.5.fc.0.bias", "features.5.fc.2.weight", "features.5.fc.2.bias", "features.6.downsample.1.weight", "features.6.downsample.1.bias", "features.6.downsample.1.running_mean", "features.6.downsample.1.running_var", "features.6.downsample.2.weight", "features.6.conv.0.weight", "features.6.conv.1.weight", "features.6.conv.1.bias", "features.6.conv.1.running_mean", "features.6.conv.1.running_var", "features.6.conv.3.weight", "features.6.conv.4.weight", "features.6.conv.4.bias", "features.6.conv.4.running_mean", "features.6.conv.4.running_var", "features.6.conv.6.weight", "features.6.conv.7.weight", "features.6.conv.7.bias", "features.6.conv.7.running_mean", "features.6.conv.7.running_var", "features.7.conv.0.weight", "features.7.conv.1.weight", "features.7.conv.1.bias", "features.7.conv.1.running_mean", "features.7.conv.1.running_var", "features.7.conv.3.weight", "features.7.conv.4.weight", "features.7.conv.4.bias", "features.7.conv.4.running_mean", "features.7.conv.4.running_var", "features.7.conv.6.weight", "features.7.conv.7.weight", "features.7.conv.7.bias", "features.7.conv.7.running_mean", "features.7.conv.7.running_var", "features.8.conv.0.weight", "features.8.conv.1.weight", "features.8.conv.1.bias", "features.8.conv.1.running_mean", "features.8.conv.1.running_var", "features.8.conv.3.weight", "features.8.conv.4.weight", "features.8.conv.4.bias", "features.8.conv.4.running_mean", "features.8.conv.4.running_var", "features.8.conv.6.weight", "features.8.conv.7.weight", "features.8.conv.7.bias", "features.8.conv.7.running_mean", "features.8.conv.7.running_var", "features.9.conv.0.weight", "features.9.conv.1.weight", "features.9.conv.1.bias", "features.9.conv.1.running_mean", "features.9.conv.1.running_var", "features.9.conv.3.weight", "features.9.conv.4.weight", "features.9.conv.4.bias", "features.9.conv.4.running_mean", "features.9.conv.4.running_var", "features.9.conv.6.weight", "features.9.conv.7.weight", "features.9.conv.7.bias", "features.9.conv.7.running_mean", "features.9.conv.7.running_var", "features.10.conv.0.weight", "features.10.conv.1.weight", "features.10.conv.1.bias", "features.10.conv.1.running_mean", "features.10.conv.1.running_var", "features.10.conv.3.weight", "features.10.conv.4.weight", "features.10.conv.4.bias", "features.10.conv.4.running_mean", "features.10.conv.4.running_var", "features.10.conv.6.weight", "features.10.conv.7.weight", "features.10.conv.7.bias", "features.10.conv.7.running_mean", "features.10.conv.7.running_var", "features.11.conv.0.weight", "features.11.conv.1.weight", "features.11.conv.1.bias", "features.11.conv.1.running_mean", "features.11.conv.1.running_var", "features.11.conv.3.weight", "features.11.conv.4.weight", "features.11.conv.4.bias", "features.11.conv.4.running_mean", "features.11.conv.4.running_var", "features.11.conv.6.weight", "features.11.conv.7.weight", "features.11.conv.7.bias", "features.11.conv.7.running_mean", "features.11.conv.7.running_var", "features.12.fc.0.weight", "features.12.fc.0.bias", "features.12.fc.2.weight", "features.12.fc.2.bias", "features.13.downsample.1.weight", "features.13.downsample.1.bias", "features.13.downsample.1.running_mean", "features.13.downsample.1.running_var", "features.13.downsample.2.weight", "features.13.conv.0.weight", "features.13.conv.1.weight", "features.13.conv.1.bias", "features.13.conv.1.running_mean", "features.13.conv.1.running_var", "features.13.conv.3.weight", "features.13.conv.4.weight", "features.13.conv.4.bias", "features.13.conv.4.running_mean", "features.13.conv.4.running_var", "features.13.conv.6.weight", "features.13.conv.7.weight", "features.13.conv.7.bias", "features.13.conv.7.running_mean", "features.13.conv.7.running_var", "features.14.conv.0.weight", "features.14.conv.1.weight", "features.14.conv.1.bias", "features.14.conv.1.running_mean", "features.14.conv.1.running_var", "features.14.conv.3.weight", "features.14.conv.4.weight", "features.14.conv.4.bias", "features.14.conv.4.running_mean", "features.14.conv.4.running_var", "features.14.conv.6.weight", "features.14.conv.7.weight", "features.14.conv.7.bias", "features.14.conv.7.running_mean", "features.14.conv.7.running_var", "features.15.conv.0.weight", "features.15.conv.1.weight", "features.15.conv.1.bias", "features.15.conv.1.running_mean", "features.15.conv.1.running_var", "features.15.conv.3.weight", "features.15.conv.4.weight", "features.15.conv.4.bias", "features.15.conv.4.running_mean", "features.15.conv.4.running_var", "features.15.conv.6.weight", "features.15.conv.7.weight", "features.15.conv.7.bias", "features.15.conv.7.running_mean", "features.15.conv.7.running_var", "features.16.fc.0.weight", "features.16.fc.0.bias", "features.16.fc.2.weight", "features.16.fc.2.bias", "final_DW.0.weight".
Unexpected key(s) in state_dict: "module.features.0.0.weight", "module.features.0.1.weight", "module.features.0.1.bias", "module.features.0.1.running_mean", "module.features.0.1.running_var", "module.features.0.1.num_batches_tracked", "module.features.1.downsample.1.weight", "module.features.1.downsample.1.bias", "module.features.1.downsample.1.running_mean", "module.features.1.downsample.1.running_var", "module.features.1.downsample.1.num_batches_tracked", "module.features.1.downsample.2.weight", "module.features.1.conv.0.weight", "module.features.1.conv.1.weight", "module.features.1.conv.1.bias", "module.features.1.conv.1.running_mean", "module.features.1.conv.1.running_var", "module.features.1.conv.1.num_batches_tracked", "module.features.1.conv.3.weight", "module.features.1.conv.4.weight", "module.features.1.conv.4.bias", "module.features.1.conv.4.running_mean", "module.features.1.conv.4.running_var", "module.features.1.conv.4.num_batches_tracked", "module.features.2.fc.0.weight", "module.features.2.fc.0.bias", "module.features.2.fc.2.weight", "module.features.2.fc.2.bias", "module.features.3.downsample.1.weight", "module.features.3.downsample.1.bias", "module.features.3.downsample.1.running_mean", "module.features.3.downsample.1.running_var", "module.features.3.downsample.1.num_batches_tracked", "module.features.3.downsample.2.weight", "module.features.3.conv.0.weight", "module.features.3.conv.1.weight", "module.features.3.conv.1.bias", "module.features.3.conv.1.running_mean", "module.features.3.conv.1.running_var", "module.features.3.conv.1.num_batches_tracked", "module.features.3.conv.3.weight", "module.features.3.conv.4.weight", "module.features.3.conv.4.bias", "module.features.3.conv.4.running_mean", "module.features.3.conv.4.running_var", "module.features.3.conv.4.num_batches_tracked", "module.features.3.conv.6.weight", "module.features.3.conv.7.weight", "module.features.3.conv.7.bias", "module.features.3.conv.7.running_mean", "module.features.3.conv.7.running_var", "module.features.3.conv.7.num_batches_tracked", "module.features.4.conv.0.weight", "module.features.4.conv.1.weight", "module.features.4.conv.1.bias", "module.features.4.conv.1.running_mean", "module.features.4.conv.1.running_var", "module.features.4.conv.1.num_batches_tracked", "module.features.4.conv.3.weight", "module.features.4.conv.4.weight", "module.features.4.conv.4.bias", "module.features.4.conv.4.running_mean", "module.features.4.conv.4.running_var", "module.features.4.conv.4.num_batches_tracked", "module.features.4.conv.6.weight", "module.features.4.conv.7.weight", "module.features.4.conv.7.bias", "module.features.4.conv.7.running_mean", "module.features.4.conv.7.running_var", "module.features.4.conv.7.num_batches_tracked", "module.features.5.fc.0.weight", "module.features.5.fc.0.bias", "module.features.5.fc.2.weight", "module.features.5.fc.2.bias", "module.features.6.downsample.1.weight", "module.features.6.downsample.1.bias", "module.features.6.downsample.1.running_mean", "module.features.6.downsample.1.running_var", "module.features.6.downsample.1.num_batches_tracked", "module.features.6.downsample.2.weight", "module.features.6.conv.0.weight", "module.features.6.conv.1.weight", "module.features.6.conv.1.bias", "module.features.6.conv.1.running_mean", "module.features.6.conv.1.running_var", "module.features.6.conv.1.num_batches_tracked", "module.features.6.conv.3.weight", "module.features.6.conv.4.weight", "module.features.6.conv.4.bias", "module.features.6.conv.4.running_mean", "module.features.6.conv.4.running_var", "module.features.6.conv.4.num_batches_tracked", "module.features.6.conv.6.weight", "module.features.6.conv.7.weight", "module.features.6.conv.7.bias", "module.features.6.conv.7.running_mean", "module.features.6.conv.7.running_var", "module.features.6.conv.7.num_batches_tracked", "module.features.7.conv.0.weight", "module.features.7.conv.1.weight", "module.features.7.conv.1.bias", "module.features.7.conv.1.running_mean", "module.features.7.conv.1.running_var", "module.features.7.conv.1.num_batches_tracked", "module.features.7.conv.3.weight", "module.features.7.conv.4.weight", "module.features.7.conv.4.bias", "module.features.7.conv.4.running_mean", "module.features.7.conv.4.running_var", "module.features.7.conv.4.num_batches_tracked", "module.features.7.conv.6.weight", "module.features.7.conv.7.weight", "module.features.7.conv.7.bias", "module.features.7.conv.7.running_mean", "module.features.7.conv.7.running_var", "module.features.7.conv.7.num_batches_tracked", "module.features.8.conv.0.weight", "module.features.8.conv.1.weight", "module.features.8.conv.1.bias", "module.features.8.conv.1.running_mean", "module.features.8.conv.1.running_var", "module.features.8.conv.1.num_batches_tracked", "module.features.8.conv.3.weight", "module.features.8.conv.4.weight", "module.features.8.conv.4.bias", "module.features.8.conv.4.running_mean", "module.features.8.conv.4.running_var", "module.features.8.conv.4.num_batches_tracked", "module.features.8.conv.6.weight", "module.features.8.conv.7.weight", "module.features.8.conv.7.bias", "module.features.8.conv.7.running_mean", "module.features.8.conv.7.running_var", "module.features.8.conv.7.num_batches_tracked", "module.features.9.conv.0.weight", "module.features.9.conv.1.weight", "module.features.9.conv.1.bias", "module.features.9.conv.1.running_mean", "module.features.9.conv.1.running_var", "module.features.9.conv.1.num_batches_tracked", "module.features.9.conv.3.weight", "module.features.9.conv.4.weight", "module.features.9.conv.4.bias", "module.features.9.conv.4.running_mean", "module.features.9.conv.4.running_var", "module.features.9.conv.4.num_batches_tracked", "module.features.9.conv.6.weight", "module.features.9.conv.7.weight", "module.features.9.conv.7.bias", "module.features.9.conv.7.running_mean", "module.features.9.conv.7.running_var", "module.features.9.conv.7.num_batches_tracked", "module.features.10.conv.0.weight", "module.features.10.conv.1.weight", "module.features.10.conv.1.bias", "module.features.10.conv.1.running_mean", "module.features.10.conv.1.running_var", "module.features.10.conv.1.num_batches_tracked", "module.features.10.conv.3.weight", "module.features.10.conv.4.weight", "module.features.10.conv.4.bias", "module.features.10.conv.4.running_mean", "module.features.10.conv.4.running_var", "module.features.10.conv.4.num_batches_tracked", "module.features.10.conv.6.weight", "module.features.10.conv.7.weight", "module.features.10.conv.7.bias", "module.features.10.conv.7.running_mean", "module.features.10.conv.7.running_var", "module.features.10.conv.7.num_batches_tracked", "module.features.11.conv.0.weight", "module.features.11.conv.1.weight", "module.features.11.conv.1.bias", "module.features.11.conv.1.running_mean", "module.features.11.conv.1.running_var", "module.features.11.conv.1.num_batches_tracked", "module.features.11.conv.3.weight", "module.features.11.conv.4.weight", "module.features.11.conv.4.bias", "module.features.11.conv.4.running_mean", "module.features.11.conv.4.running_var", "module.features.11.conv.4.num_batches_tracked", "module.features.11.conv.6.weight", "module.features.11.conv.7.weight", "module.features.11.conv.7.bias", "module.features.11.conv.7.running_mean", "module.features.11.conv.7.running_var", "module.features.11.conv.7.num_batches_tracked", "module.features.12.fc.0.weight", "module.features.12.fc.0.bias", "module.features.12.fc.2.weight", "module.features.12.fc.2.bias", "module.features.13.downsample.1.weight", "module.features.13.downsample.1.bias", "module.features.13.downsample.1.running_mean", "module.features.13.downsample.1.running_var", "module.features.13.downsample.1.num_batches_tracked", "module.features.13.downsample.2.weight", "module.features.13.conv.0.weight", "module.features.13.conv.1.weight", "module.features.13.conv.1.bias", "module.features.13.conv.1.running_mean", "module.features.13.conv.1.running_var", "module.features.13.conv.1.num_batches_tracked", "module.features.13.conv.3.weight", "module.features.13.conv.4.weight", "module.features.13.conv.4.bias", "module.features.13.conv.4.running_mean", "module.features.13.conv.4.running_var", "module.features.13.conv.4.num_batches_tracked", "module.features.13.conv.6.weight", "module.features.13.conv.7.weight", "module.features.13.conv.7.bias", "module.features.13.conv.7.running_mean", "module.features.13.conv.7.running_var", "module.features.13.conv.7.num_batches_tracked", "module.features.14.conv.0.weight", "module.features.14.conv.1.weight", "module.features.14.conv.1.bias", "module.features.14.conv.1.running_mean", "module.features.14.conv.1.running_var", "module.features.14.conv.1.num_batches_tracked", "module.features.14.conv.3.weight", "module.features.14.conv.4.weight", "module.features.14.conv.4.bias", "module.features.14.conv.4.running_mean", "module.features.14.conv.4.running_var", "module.features.14.conv.4.num_batches_tracked", "module.features.14.conv.6.weight", "module.features.14.conv.7.weight", "module.features.14.conv.7.bias", "module.features.14.conv.7.running_mean", "module.features.14.conv.7.running_var", "module.features.14.conv.7.num_batches_tracked", "module.features.15.conv.0.weight", "module.features.15.conv.1.weight", "module.features.15.conv.1.bias", "module.features.15.conv.1.running_mean", "module.features.15.conv.1.running_var", "module.features.15.conv.1.num_batches_tracked", "module.features.15.conv.3.weight", "module.features.15.conv.4.weight", "module.features.15.conv.4.bias", "module.features.15.conv.4.running_mean", "module.features.15.conv.4.running_var", "module.features.15.conv.4.num_batches_tracked", "module.features.15.conv.6.weight", "module.features.15.conv.7.weight", "module.features.15.conv.7.bias", "module.features.15.conv.7.running_mean", "module.features.15.conv.7.running_var", "module.features.15.conv.7.num_batches_tracked", "module.features.16.fc.0.weight", "module.features.16.fc.0.bias", "module.features.16.fc.2.weight", "module.features.16.fc.2.bias", "module.final_DW.0.weight".I met this problem too. I think maybe the definition of the model in FeatherNet.py is different from the model in the pretrain model.
This solved this problem. Run <model = nn.DataParallel(model) & cudnn.benchmark = True > before <load_state_dict>. But i am not exactly sure it is the correct solution. you can try it.
checkpoint = torch.load('checkpoints/FeatherNetB_bs32/_47_best.pth.tar', map_location='cpu')
model = nn.DataParallel(model)
cudnn.benchmark = True
model.load_state_dict(checkpoint['state_dict'])
would you like to share your datatset to me? thank you
Are the given weights 'mobilenet_v2.pth.tar' can be used when test phase?
No 'epoch' ,'best_prec1',key
And the layers in state_dict are all missed a 'module.'
the shape of module.classifier.1.weight is also mismatched