Open yxchng opened 6 months ago
I never got around to hooking that up to timm based models. It's possible, but requires a bit of extra code to handle the resizing properly....
@rwightman any update on this?
@yxchng I actually do have some work in progress right now in timm
to address https://github.com/huggingface/pytorch-image-models/issues/2190 (adding a set_input_size()
) fn to vit, that would allow this to work, I don't think I can rely on the @ pretrained load resize that's already there for timm weights
setting force_image_size to 672 still give model with default 336 input size