yolov4-pytorch 置信度损失一直都比较高

置信度损失一直都比较高

Open Willianwatch opened this issue 2 years ago • 6 comments

你好，请问下置信度损失一直都比较高有可能是什么原因呢？

Mar 13 '22 02:03 Willianwatch

目标比较难检测？

Mar 14 '22 16:03 bubbliiiing

我也不知道为啥，我换了yolov4_tiny之后就好了😂

Mar 15 '22 00:03 Willianwatch

我还有一个问题，我使用单块GPU，但是我的数据增强部分比较慢，这是我程序的瓶颈，想使用多个节点的cpu做数据增强这一部分，大佬知道该怎么做吗？

Mar 15 '22 00:03 Willianwatch

@Willianwatch 多个节点的cpu/gpu应该都可以试试huggingface的accelerate或者horovod，deepspeed这些的把，之前我尝试部署过单机多卡的分布式训练，然后在权重保存和日志打印那里遇到点问题就没再继续了

Mar 15 '22 01:03 XuecWu

你只有单块GPU的话应该是设置num_works即可。

Mar 15 '22 02:03 bubbliiiing

没必要多节点，分布式的话需要你有多卡，你没有多卡就是num_work，哪怕是分布式，也是一个节点多个num_works

Mar 15 '22 02:03 bubbliiiing