Cannot load pretrain metafg_2_inat21_384

dqshuai / MetaFormer

A PyTorch implementation of "MetaFormer: A Unified Meta Framework for Fine-Grained Recognition". A reference PyTorch implementation of “CoAtNet: Marrying Convolution and Attention for All Data Sizes”

MIT License

226 stars 40 forks source link

Cannot load pretrain metafg_2_inat21_384 #32

Open DuyMinhLe13 opened 1 year ago

DuyMinhLe13 commented 1 year ago

I use default config for MetaFG_2 but the errors in load_state_dict occured. Could you provide config of MetaFG_2_384 ?

sun-jiao commented 1 year ago

Config file for metafg_2_inat21_384:

DATA:
  IMG_SIZE: 384
  ADD_META: True
MODEL:
  TYPE: MetaFG
  NAME: MetaFG_meta_2
  EXTRA_TOKEN_NUM: 3
  META_DIMS: [ 4, 3 ]
  NUM_CLASSES: 10000

I am not a member of their team. I just tried it out.