模型输入是否需要渐变?

时间:2019-07-18 06:25:06

标签: image-processing machine-learning computer-vision pytorch gradient-descent

在pytorch中,什么时候需要将输入张量的require_grad设置为True

1 个答案:

答案 0 :(得分:2)

通常,输入是固定的-我们不更改图像,我们仅从固定的输入图像推断标签/输出。
由于它们是固定的-无需计算输入的斜率,只需计算可训练的参数即可。

话虽如此,在某些情况下您想更改输入。例如,当您要可视化要素时。参见例如this nice post