Method _forward_impl

modules/VIPTRv1.py:500–504

(self, x, relative_pos_enc=None)

            drop_path) if drop_path > 0. else nn.Identity()

    def _forward_impl(self, x, relative_pos_enc=None):
        # x = x + self.pos_embed(x)
        x = x + self.drop_path(self.token_mixer(self.norm1(x), relative_pos_enc))
        x = x + self.drop_path(self.mlp(self.norm2(x)))
        return x

    def forward(self, x, relative_pos_enc=None):
        if self.grad_checkpoint and x.requires_grad: