Loss计算花费时间过多。

bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码，可以用于训练自己的模型。

MIT License

780 stars 185 forks source link

Loss计算花费时间过多。 #95

Open Wu1332451126 opened 2 years ago

Wu1332451126 commented 2 years ago

自己训练coco。发现GPU利用率一直波动，之前训练imageNet的GPU可以吃满。经过测试，发现Loss计算是限制瓶颈，这样只能换CPU了吗。CPU：Intel(R) Xeon(R) Silver 4214 CPU 2.20GHz GPU：3090 0ecd8b6379ac52f025c24b3cd3a3865

bubbliiiing commented 2 years ago

我的损失里面包含了匹配的部分，比较久，可以参考yolov5的损失改改，就会快乐