Method patch_size

examples/common/model_loader.py:153–154

(self)

    @property
    def patch_size(self):
        return self._patch_size

    def forward(self, x: torch.Tensor, *args, **kwargs):
        out_h = x.shape[-2] // self._patch_size