How to train this model? the command line and needed params?

MengLcool / AdaViT

[CVPR-22] This is the official implementation of the paper "Adavit: Adaptive vision transformers for efficient image recognition".

49 stars 8 forks source link

How to train this model? the command line and needed params? #5

Open Carol-lyh opened 2 years ago

Carol-lyh commented 2 years ago

I download the model ckpt, but there is no .pth file. Instead, it's .pkl file. When loading this .pkl file into the model, the following error was encountered: Unpickling error: A load persistent id instruction was encountered but no persistent_load function was specified. Could you tell me how to load the ckpt?

Carol-lyh commented 2 years ago

I've solved the above problem! And the inference Acc is 81.1%. I want to know the command line for training this model. Thank you very much!

marina-neseem commented 2 years ago

Hi, How did you solve the above problem? I am facing the same problem

Carol-lyh commented 2 years ago

DO NOT extract the .tar file, it's not a compressed file. JUST load the .tar file as the ckpt.

yu868 commented 2 years ago

请问作者能够释放训练超参数吗？只是一个推理的超参很难让人信服啊

W4ngH4o commented 2 years ago

Do you know what is the flops_dict.pth? Thank you