search

Sense-X / UniFormer

[ICLR2022] official implementation of UniFormer
Apache License 2.0
812 stars 111 forks source link

image_classification_model(Large resolution fine-tuning (384x384)) #128

Closed wanzhen666 closed 7 months ago

wanzhen666 commented 7 months ago

Hello, is the dim size 64 for the pre-trained model of 384×384 size in image classification? 1705546116891

wanzhen666 commented 7 months ago

@Andy1621

Andy1621 commented 7 months ago

Hi, for these models, they use dim=32, you can find the related model shape in run.sh.

https://github.com/Sense-X/UniFormer/blob/52a0415e4e0dda880fab8ffbc710e4d8e171149d/image_classification/token_labeling/exp/uniformer_small_plus_tl_384/run.sh#L4

wanzhen666 commented 7 months ago

I see. Thank you very much