2020
09-23
09-23
在pytorch中实现只让指定变量向后传播梯度
pytorch中如何只让指定变量向后传播梯度?(或者说如何让指定变量不参与后向传播?)有以下公式,假如要让L对xvar求导:(1)中,L对xvar的求导将同时计算out1部分和out2部分;(2)中,L对xvar的求导只计算out2部分,因为out1的requires_grad=False;(3)中,L对xvar的求导只计算out1部分,因为out2的requires_grad=False;验证如下:#!/usr/bin/envpython2#-*-coding:utf-8-*-"""CreatedonWedMay2310:02:042018@auth...
继续阅读 >