ast icon indicating copy to clipboard operation
ast copied to clipboard

Unknown model (vit_deit_base_distilled_patch16_384)

Open BaronWang0130 opened this issue 1 year ago • 3 comments

捕获

BaronWang0130 avatar Aug 02 '22 15:08 BaronWang0130

Is it the reason for timm version?

BaronWang0130 avatar Aug 02 '22 16:08 BaronWang0130

Hi there,

It seems to be a timm version problem, please use timm==0.4.5. In Colab, try using something like !pip install timm==0.4.5.

Nevertheless, the code should report an error when the timm version is not consistent.https://github.com/YuanGongND/ast/blob/d7d8b4b8e06cdaeb6c843cdb38794c1c7692234c/src/models/ast_models.py#L50. Can you check timm.__version__?

-Yuan

YuanGongND avatar Aug 02 '22 16:08 YuanGongND

It works now. Thanks.

BaronWang0130 avatar Aug 03 '22 13:08 BaronWang0130