fix name

adb99303 · 悟、 · Walter · 5604a13f · adb99303
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppcls/arch/backbone/model_zoo/foundation_vit.py ppcls/arch/backbone/model_zoo/foundation_vit.py +1 -1

未找到文件。
--- 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': [],
    }