huawei-noah / Efficient-AI-Backbones

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

关于VIG,训练 #195

Closed Jasookii closed 1 year ago

Jasookii commented 1 year ago

作者你好,我用8张1080Ti进行训练vig_ti模型到125个epoch后,结果仍然较差,请问这是正常现象吗,还是显卡问题? VIG

Jasookii commented 1 year ago

使用预训练模型进行测试也是同样的问题,是我的显卡损坏了吗? test

iamhankai commented 1 year ago

可以看看你前几个epoch的日志吗

Jasookii commented 1 year ago

截屏2023-04-11 20 09 43

Jasookii commented 1 year ago

截屏2023-04-11 20 09 53

Jasookii commented 1 year ago

截屏2023-04-11 20 10 03

Jasookii commented 1 year ago

截屏2023-04-11 20 10 12

Jasookii commented 1 year ago

截屏2023-04-11 20 10 22

Jasookii commented 1 year ago

我在另一个显卡上使用您提供的预训练参数进行测试,结果仍然很差。我理解您提供的预训练参数是在imagenet1k上训练好的,是可以直接在该训练集上测试,还是需要在您提供的预训练参数的基础上进行训练吗(我是:(新手,还请您指教 截屏2023-04-11 22 28 51

Jasookii commented 1 year ago

imagenet1k的测试集合怎么没有分文件夹呀,5555,现在可以复现了,感谢您优秀的工作。

iamhankai commented 1 year ago

好吧,我看你train loss正常,test loss不正常,所以应该是test set有点问题