JXingZhao / EGNet

EGNet:Edge Guidance Network for Salient Object Detection (ICCV 2019)
311 stars 62 forks source link

再次训练提供的训练好的模型报错 #25

Closed LinLi-DL closed 4 years ago

LinLi-DL commented 4 years ago

您好, 我想用你训练好的模型再训练一下我自己的一个小数据集,但是直接报错了。Error: Error(s) in loading state_dict for ResNet: Missing key(s) in state_dict: "conv1.weight",.......

slover中的代码是这句: self.net_bone.base.load_state_dict(torch.load(self.config.resnet))

在网上找了一些办法,但是都没解决,请帮忙指导下,谢谢。