bubbliiiing / yolov4-tiny-pytorch

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

大佬你好,我在阅读代码中有一个疑问。 #100

Closed luzhouWH closed 1 year ago

luzhouWH commented 1 year ago

在nets的CSPdarknet53_tiny.py的class Resblock_body(nn.Module)其中的def forward(self, x)。 您使用route = x的复制,但是在python中这样好像是给的x的地址,后续x变化的话,route 也会跟着变化,达不到残差的效果了。 请问这里需不需要改写成深度复制?

bubbliiiing commented 1 year ago

你可以看看route 和x的值0 0,应该没有这个问题

luzhouWH commented 1 year ago

刚刚看了,真的不一样。非常感谢大佬。