ardeal/yolo_nano

训练验证时间太长?

JingyuLi-code opened this issue · 3 comments

训练时 batchsize设置为32,rtx2080ti 训练验证时 batchsize默认为1 验证的时间长达 十几分钟,训练的时候使用了 GPU, 什么问题导致??
验证计算MAP完成 抛出以下错误:
Traceback (most recent call last):
File "train_yolonano.py", line 149, in
ap_table += [[c, class_names[c], "%.5f" % AP[i]]]
IndexError: list index out of range
谢谢!

1) 2080Ti, 如果batchsize=32,好像显存不够用。我的也是2080Ti,我设置的bathsize=4,好像设置batchsize=8也可以。16/32,我就没有尝试过。
2)149行出错,原因是evaluate返回的AP可能是空的。AP返回空的原因,可能是训练不好,导致evaluate效果很差。

嗯,对的 batchsize设置16,32好像不行,那是什么原因导致训练时验证时间那么长呢? 十几分钟? 我只有1500张验证图片,在YOLOv3验证时只需要一分钟不到。

我evaluate 时间很短,不需要十几分钟。
你是不是修改了什么,或者,你验证的样本数量是不是太多了?或者,你验证期间,是不是imshow了?
你应该调试一下看看,到底是哪里慢。