torch 梯度无法传导的一个可能的原因

问题描述

frame2 = (frame + noise).clamp(MIN, MAX)

noise 需要并且设置了梯度，frame 是一个 C*H*W 的普通的帧，但是 frame2 无法梯度传导回 noise 上。

一些参数：

1. frame.is_leaf 为 True, frame.requires_grad 为 False
2. noise.is_leaf 为 True, noise.requires_grad 为 True
3. 但是，frame2.is_leaf 为 True, frame2.requires_grad 为 False

解决方案

处在 torch.no_grad() 环境之中，导致梯度切断了，设置 torch.enable_grad() 来恢复梯度传输。

···