search

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。
MIT License
786 stars 184 forks source link

训练过程资源记载缓慢 #117

Open PeterLu8888 opened 11 months ago

PeterLu8888 commented 11 months ago

我发现训练过程很慢。测试的只是很小的图片集（300张）， 是卡在fit_one_epoch方法的for iteration, batch in enumerate(gen)这句 这时GPU没有消耗，只是在不断创建新的线程。

请问大佬是不是因为资源加载的问题，有办法优化么？