search

Lavender105 / DFF

Code for Dynamic Feature Fusion for Semantic Edge Detection https://arxiv.org/abs/1902.09104
MIT License
220 stars 51 forks source link

RuntimeError: Error(s) in loading state_dict for DFF #25

Open visionxyz opened 3 years ago

visionxyz commented 3 years ago

RuntimeError: Error(s) in loading state_dict for DFF: Missing key(s) in state_dict: "ada_learner.conv1.0.weight", "ada_learner.conv1.0.bias", "ada_learner.conv1.1.weight", "ada_learner.conv1.1.bias", "ada_learner.conv1.1.running_mean", "ada_learner.conv1.1.running_var", "ada_learner.conv2.0.weight", "ada_learner.conv2.0.bias", "ada_learner.conv2.1.weight", "ada_learner.conv2.1.bias", "ada_learner.conv2.1.running_mean", "ada_learner.conv2.1.running_var", "ada_learner.conv3.0.weight", "ada_learner.conv3.0.bias", "ada_learner.conv3.1.weight", "ada_learner.conv3.1.bias", "ada_learner.conv3.1.running_mean", "ada_learner.conv3.1.running_var", "side5_w.0.weight", "side5_w.0.bias", "side5_w.1.weight", "side5_w.1.bias", "side5_w.1.running_mean", "side5_w.1.running_var", "side5_w.2.weight". Unexpected key(s) in state_dict: "EW1.conv1.0.weight", "EW1.conv1.0.bias", "EW1.conv1.1.weight", "EW1.conv1.1.bias", "EW1.conv1.1.running_mean", "EW1.conv1.1.running_var", "EW1.conv1.1.num_batches_tracked", "EW1.conv2.0.weight", "EW1.conv2.0.bias", "EW1.conv2.1.weight", "EW1.conv2.1.bias", "EW1.conv2.1.running_mean", "EW1.conv2.1.running_var", "EW1.conv2.1.num_batches_tracked", "EW1.conv3.0.weight", "EW1.conv3.0.bias", "EW1.conv3.1.weight", "EW1.conv3.1.bias", "EW1.conv3.1.running_mean", "EW1.conv3.1.running_var", "EW1.conv3.1.num_batches_tracked", "side5_ew.0.weight", "side5_ew.0.bias", "side5_ew.1.weight", "side5_ew.1.bias", "side5_ew.1.running_mean", "side5_ew.1.running_var", "side5_ew.1.num_batches_tracked", "side5_ew.2.weight".

Thanks for your work! I am trying to use this model in my work, but facing such a problem, maybe I am not using this in a proper way? Is there any way to help me with this bug, thanks!