huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

vig关于train.py模版

Jasookii opened this issue · comments

commented

截屏2023-04-24 19 51 26

中间这条的测试结果的Loss为什么这么大呀
commented

截屏2023-04-24 19 54 47

并且有时偶尔会出现nan的情况,我应该怎么办呢

这是哪个模型

commented

我将VIG进行了一定的修改,想问下上述情况是哪里出了问题?

出现nan就代表模型已经出现问题了,检查一下模型weight是不是过大或者过小了

commented

hankai大神你好呀,我设置batch_size=128,我发现测试时,最后一个batch的数据量远小于128,并且loss非常大,这正常吗