How the load the pretrained model on classification when trained on segmentation?

[liyiersan]

The positional embedding is related to the input size, is it possible to load the pretrained models on classification? By the way, it seems that torch.unfold is faster than torch.roll on GPUs.