search

linksense / LightNetPlusPlus

LightNet++: Boosted Light-weighted Networks for Real-time Semantic Segmentation
https://ansleliu.github.io/LightNet.html
MIT License
243 stars 54 forks source link

Missing keys in state dict #3

Closed barnjamin closed 3 years ago

barnjamin commented 5 years ago

Only the deepdrive checkpoint file seems to have the usm.norm_act.weight

...
> # ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ #
> 1. Setting up Model...
Traceback (most recent call last):
  File "deploy/cityscapes/evaluation/ss.py", line 253, in <module>
    model.load_state_dict(pre_weight)
  File "/home/ben/LightNetPlusPlus/env/lib/python3.6/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DataParallel:
    Missing key(s) in state_dict: "module.usm.norm_act.weight", "module.usm.norm_act.bias", "module.usm.norm_act.running_mean", "module.usm.norm_act.running_var". 

(env) ben@mercury:~/LightNetPlusPlus$ grep -r norm_act.weight ./*
Binary file ./checkpoint/cityscapes_mobilenetv2plus_x1.0.pkl matches
Binary file ./checkpoint/cityscapes_shufflenetv2plus_x0.5.pkl matches
Binary file ./checkpoint/cityscapes_shufflenetv2plus_x1.0.pkl matches
Binary file ./checkpoint/deepdrive_shufflenetv2plus_x1.0.pkl matches

(env) ben@mercury:~/LightNetPlusPlus$ grep -r usm.norm_act.weight ./*
Binary file ./checkpoint/deepdrive_shufflenetv2plus_x1.0.pkl matches