diff --git a/ppcls/arch/backbone/model_zoo/foundation_vit.py b/ppcls/arch/backbone/model_zoo/foundation_vit.py index 670efaabed15cafe5993577b1fe45bb7aaa4776e..66000516dcfffd8fd6bfd68b8fa5998dcc5fda37 100644 --- a/ppcls/arch/backbone/model_zoo/foundation_vit.py +++ b/ppcls/arch/backbone/model_zoo/foundation_vit.py @@ -151,7 +151,7 @@ _EVA_diff = { 'remove_abs_pos_emb': [], 'replace_mlp_GELU': [], 'head': { - 'fc_norm': ['huge_vit_patch14'], + 'fc_norm': ['vit_huge_patch14'], 'return_all_tokens': [], 'return_patch_tokens': [], }