Jane_O的博客

原创 pytorch中的.detach()

在GAN的学习中，一个UP主讲解的gan代码时提到：“训练鉴别器，把梯度截断，不让其传到生成器中”，然后用到了.detach（）函数，发现自己有点懵B以下是对.detach()的部分理解目的：1.希望保持部分网络参数不变，只对其中一部分的参数进行调整；2.智训练部分分支网络，并不让其梯度对主网络的地图造成影响，此时我们用detach()函数 来切断一些分支的反向传播。tensor.detach()：返回一个新的tensor，从当前计算图中分离下来的，但是仍指向原变量的存放位置,不同.