关于ViG验证的问题

[Zsl0320]

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

[Zsl0320]

我找到问题所在了，

我需要额外加上state_dict = state_dict['state_dict']才可以，我没有更改train.py。为什么会这样呢

[iamhankai]

ok，那就是train.py代码写错了