search

lmbxmu / 1xN

Pytorch implementation of TPAMI 2022 -- 1xN Pattern for Pruning Convolutional Neural Networks
https://arxiv.org/abs/2105.14713
43 stars 4 forks source link

about load pruned model #1

Open 523997931 opened 3 years ago

523997931 commented 3 years ago

when I try to load the pruned model, it report error RuntimeError: Error(s) in loading state_dict for MobileNetV3: Missing key(s) in state_dict: "features.1.conv.4.bias", "features.2.conv.0.bias", "features.2.conv.7.bias", "features.3.conv.0.bias", "features.3.conv.7.bias", "features.4.conv.0.bias", "features.4.conv.7.bias", "features.5.conv.0.bias", "features.5.conv.7.bias", "features.6.conv.0.bias", "features.6.conv.7.bias", "features.7.conv.0.bias", "features.7.conv.7.bias", "features.8.conv.0.bias", "features.8.conv.7.bias", "features.9.conv.0.bias", "features.9.conv.7.bias", "features.10.conv.0.bias", "features.10.conv.7.bias", "features.11.conv.0.bias", "features.11.conv.7.bias", "features.12.conv.0.bias", "features.12.conv.7.bias", "features.13.conv.0.bias", "features.13.conv.7.bias", "features.14.conv.0.bias", "features.14.conv.7.bias", "features.15.conv.0.bias", "features.15.conv.7.bias", "conv.0.bias".

lmbxmu commented 3 years ago

when I try to load the pruned model, it report error RuntimeError: Error(s) in loading state_dict for MobileNetV3: Missing key(s) in state_dict: "features.1.conv.4.bias", "features.2.conv.0.bias", "features.2.conv.7.bias", "features.3.conv.0.bias", "features.3.conv.7.bias", "features.4.conv.0.bias", "features.4.conv.7.bias", "features.5.conv.0.bias", "features.5.conv.7.bias", "features.6.conv.0.bias", "features.6.conv.7.bias", "features.7.conv.0.bias", "features.7.conv.7.bias", "features.8.conv.0.bias", "features.8.conv.7.bias", "features.9.conv.0.bias", "features.9.conv.7.bias", "features.10.conv.0.bias", "features.10.conv.7.bias", "features.11.conv.0.bias", "features.11.conv.7.bias", "features.12.conv.0.bias", "features.12.conv.7.bias", "features.13.conv.0.bias", "features.13.conv.7.bias", "features.14.conv.0.bias", "features.14.conv.7.bias", "features.15.conv.0.bias", "features.15.conv.7.bias", "conv.0.bias".

code fixed. Our apology.