Method __init__

models/unet_model_google.py:63–66

(self, dim)

class AttentiveGuide(nn.Module):
    def __init__(self, dim):
        super().__init__()
        self.norm1 = LayerNorm2d(dim)
        self.norm2 = LayerNorm2d(dim)

    def forward(self, x, g):
        return self.norm1(x) * self.norm2(g) * x