search

z1069614715 / pytorch-classifier

image classifier implement in pytoch.
MIT License
102 stars 17 forks source link

魔导，这是怎么回事啊？ #2

Open Warrior2024 opened 1 year ago

Warrior2024 commented 1 year ago

1681786987438 我删了elif 的not，权重才load进去

z1069614715 commented 1 year ago

使用了VGG是吗，我一会测试一下

z1069614715 commented 1 year ago

你使用的是VGG那个模型呢 训练还是怎么样出现的？

Warrior2024 commented 1 year ago

是的，训练出现的，vgg16，没有BN

z1069614715 commented 1 year ago

训练的时候只需要选择pretrained为True即可，weights不需要指定，我这边是正常的

Warrior2024 commented 1 year ago

魔导，那我指定weight路径，pretrained false不可以么？

z1069614715 commented 1 year ago

可以是可以的，但是这样很容易出现对不上权重，而且上面我认为没问题，这只是个判断转换而已，具体还是看输出

Warrior2024 commented 1 year ago

魔导，权重我就下载的你model_urls的权重，然后pretrained false，然后指定weight路径也能正确读到weight，但是AttributeError: 'collections.OrderedDict' object has no attribute 'state_dict'，还再麻烦您解答一下

z1069614715 commented 1 year ago

可能是个bug，这个以后看看

loganzhang4 commented 10 months ago

可以将elif这两行换成这样： elif not isinstance(state_dict, OrderedDict): state_dict = state_dict.state_dict() 这样就可以正常导入了

174326Jl commented 7 months ago

runs模型下载的链接过期了，可以重新分享一下吗？求求了