search

yizt / Grad-CAM.pytorch

pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...
Apache License 2.0
726 stars 166 forks source link

如何加载自己训练的权重文件 #6

Open zhang0391 opened 4 years ago

zhang0391 commented 4 years ago

你好，请问怎么加载自己训练的模型的权重文件啊，我在输入自己的网络和权重文件路径之后，总是会出现以下错误，其中第二行很长我省略了后面的部分。 RuntimeError: Error(s) in loading state_dict for DenseNet: Missing key(s) in state_dict: "features.conv0.weight", Unexpected key(s) in state_dict: "state_dict".

yizt commented 4 years ago

@zhang0391 保持预测模型与训练模型一致。

louis85911 commented 4 years ago

@zhang0391 https://discuss.pytorch.org/t/unexpected-key-in-state-dict-bn1-num-batches-tracked/29454/5

Dtappledoghuati commented 4 years ago

请问你是如何设置自己的网络和权重路径的，我加载模型总是出问题，非常感谢

yizt commented 4 years ago

@Dtappledoghuati 把代码和问题都贴出来把