项目介绍
在对抗学习中,我们常常需要将某一层的梯度乘上-1再进行梯度下降的操作(即梯度上升),就像【深度域适配】一、DANN与梯度反转层(GRL)详解一文中所介绍的那样。
代码实现
@tf.custom_gradient
def grad_reverse(x):y = tf.identity(x)
在对抗学习中,我们常常需要将某一层的梯度乘上-1再进行梯度下降的操作(即梯度上升),就像【深度域适配】一、DANN与梯度反转层(GRL)详解一文中所介绍的那样。
@tf.custom_gradient
def grad_reverse(x):y = tf.identity(x)