search

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.05k stars 708 forks source link

关于ViG验证的问题 #280

Open Zsl0320 opened 2 days ago

Zsl0320 commented 2 days ago

大佬，您好。目前我这里是改了一点网络去训练，但是目前遇到了问题，就是在验证预训练模型时，我使用--pretrain_path指定好路径之后，他的loss是不正常的。但是我如果--rusume这个预训练模型去评估时，他的结果就是正常的。请问这个原因是什么呢。感谢 image image

Zsl0320 commented 2 days ago

我找到问题所在了， image 我需要额外加上state_dict = state_dict['state_dict']才可以，我没有更改train.py。为什么会这样呢