The implement of GRL maybe not correct?

[Shayne-AuthMe]

The implement of GRL maybe not correct?

SSAN/networks/pub_mod.py

Line 7 in f4c2cbc

class GRL(nn.Module):

The backword function is need to inherit torch.autograd.Function like this example in pytorch as follow:

https://pytorch.org/docs/stable/autograd.html#torch.autograd.Function

A simple implement might be:

from torch.autograd import Function

class GRL(Function):

    @staticmethod
    def forward(ctx: Any, input: torch.Tensor):
        return input * 1.0

    @staticmethod
    def backward(ctx: Any, gradOutput: torch.Tensor):
        return gradOutput.neg()