WenmuZhou / DBNet.pytorch

A pytorch re-implementation of Real-time Scene Text Detection with Differentiable Binarization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问一下测试速度很慢,而且中途显存就爆了是为什么呢?训练没有问题,测试集只有500张

hellmo718 opened this issue · comments

请问一下测试速度很慢,而且中途显存就爆了是为什么呢?训练没有问题,测试集只有500张

请问你解决了吗?我训练resnet50时遇到同样的问题,val时显存一致往上涨,直到爆炸

好像是数据集有问题,有脏数据吧。测试速度慢是因为第一个epoch读取图片较慢,后面的epoch就好。

我这是显卡爆了,不知道什么原因

我这是显卡爆了,不知道什么原因

企业微信截图_16656581022521

可能是数据集的问题,可以先用icdar15跑着试下,这个是可以的,你的数据集里面可能有点问题或者啥的吧,具体我也不是很清楚,我很久没弄了

好的 谢谢