pytorch里的tricks

  • 该问题是指在默认情况下,网络在反向传播中不允许多个backward()。需要在第一个backward设置retain_graph=True。loss.backward()改为loss.backward(retain_graph=True)