Method norm2

segmentation/backbones/vit_SelfPatch.py:88–89

(self)

    @property
    def norm2(self):
        return getattr(self, self.norm2_name)

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