search

bubbliiiing / yolov4-pytorch

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

对于损失计算方式的疑惑 #343

Closed zhiyiYo closed 1 year ago

zhiyiYo commented 2 years ago

请问为啥在计算损失的时候要把特征图像一个个传进去再在外面累加损失，不能直接在 forward() 函数里面开循环来计算三个特征图的损失并求和损失吗？

bubbliiiing commented 1 year ago

理论上肯定可以呀

zhiyiYo commented 1 year ago

感谢回复