Pytorch实现梯度反转报错non-static forward method的解决办法

Air_sss

于 2021-11-17 16:19:46 发布

阅读量2.3k

点赞数 6

分类专栏： Pytorch 文章标签： python 深度学习神经网络 pytorch

本文链接：https://blog.csdn.net/coding_with_smile/article/details/121380361

版权

由于pytorch版本更新，在1.3及以上的版本中要求forward方法必须为静态方法。而在许多较早版本的代码实现中，梯度反转通常是这样写的：

from torch.autograd import Function


class GradReverse(Function):
    def __init__(self, lambd):
        self.lambd = lambd

    def forward(self, x):
        return x.view_as(x)

    def backward(self, grad_output):
        return (grad_output

最低0.47元/天解锁文章