search

SoftwareGift / FeatherNets_Face-Anti-spoofing-Attack-Detection-Challenge-CVPR2019

Code for 3rd Place Solution in Face Anti-spoofing Attack Detection Challenge @ CVPR2019,model only 0.35M!!! 1.88ms(CPU)
Other
928 stars 284 forks source link

RuntimeError: Error(s) in loading state_dict for FeatherNet: Missing key(s) in state_dict #69

Open guuboi opened 4 years ago

guuboi commented 4 years ago

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".

AnthonyF333 commented 4 years ago

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.

guuboi commented 4 years ago

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 . 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'])

lihuikenny commented 4 years ago

would you like to share your datatset to me? thank you

guuboi commented 4 years ago

Sorry, i have no raw data.

bwGong

邮箱：guuboi@163.com |

Signature is customized by Netease Mail Master

On 10/26/2019 09:54, lihuikenny wrote:

would you like to share your datatset to me? thank you

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

SoftwareGift commented 4 years ago

image

Sunzz1996 commented 4 years ago

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