Method forward

segmentation/backbones/vit_SelfPatch.py:91–94

(self, x)

        return getattr(self, self.norm2_name)

    def forward(self, x):
        x = self.attn(self.norm1(x), identity=x)
        x = self.ffn(self.norm2(x), identity=x)
        return x


@BACKBONES.register_module()