请问一下测试速度很慢,而且中途显存就爆了是为什么呢?训练没有问题,测试集只有500张
hellmo718 opened this issue · comments
hellmo718 commented
请问一下测试速度很慢,而且中途显存就爆了是为什么呢?训练没有问题,测试集只有500张
火锅女孩 commented
请问你解决了吗?我训练resnet50时遇到同样的问题,val时显存一致往上涨,直到爆炸
hellmo718 commented
好像是数据集有问题,有脏数据吧。测试速度慢是因为第一个epoch读取图片较慢,后面的epoch就好。
火锅女孩 commented
我这是显卡爆了,不知道什么原因
火锅女孩 commented
hellmo718 commented
可能是数据集的问题,可以先用icdar15跑着试下,这个是可以的,你的数据集里面可能有点问题或者啥的吧,具体我也不是很清楚,我很久没弄了
火锅女孩 commented
好的 谢谢