torch yolov3梯度发散

torch yolov3梯度发散

原因：

每次计算之前zero_grad，梯度清0，如果没有这句

梯度会累加，产生一个很大的值，后面会梯度发散。

imgs = Variable(imgs.type(Tensor))

targets = Variable(targets.type(Tensor), requires_grad=False)
optimizer.zero_grad()
loss = model(imgs, targets)
loss.backward()
optimizer.step()

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/80783617