loss1.backward(retain_graph=True) # 第一次反向传播,保留计算图
loss2.backward() # 第二次反向传播
retain_graph,requires_grad,create_graph区别
最新推荐文章于 2024-05-19 09:01:25 发布
loss1.backward(retain_graph=True) # 第一次反向传播,保留计算图
loss2.backward() # 第二次反向传播