diff --git a/ppcls/configs/ImageNet/LeViT/LeViT_128.yaml b/ppcls/configs/ImageNet/LeViT/LeViT_128.yaml index 0dabfeb8c..6683bf823 100644 --- a/ppcls/configs/ImageNet/LeViT/LeViT_128.yaml +++ b/ppcls/configs/ImageNet/LeViT/LeViT_128.yaml @@ -17,7 +17,7 @@ Global: # model architecture Arch: - name: LeViT128 + name: LeViT_128 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/LeViT/LeViT_128S.yaml b/ppcls/configs/ImageNet/LeViT/LeViT_128S.yaml index 958868fd9..3b7caf0ce 100644 --- a/ppcls/configs/ImageNet/LeViT/LeViT_128S.yaml +++ b/ppcls/configs/ImageNet/LeViT/LeViT_128S.yaml @@ -17,7 +17,7 @@ Global: # model architecture Arch: - name: LeViT128S + name: LeViT_128S # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/LeViT/LeViT_192.yaml b/ppcls/configs/ImageNet/LeViT/LeViT_192.yaml index 0ecf2c365..926d43232 100644 --- a/ppcls/configs/ImageNet/LeViT/LeViT_192.yaml +++ b/ppcls/configs/ImageNet/LeViT/LeViT_192.yaml @@ -17,7 +17,7 @@ Global: # model architecture Arch: - name: LeViT192 + name: LeViT_192 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/LeViT/LeViT_256.yaml b/ppcls/configs/ImageNet/LeViT/LeViT_256.yaml index 2acccef61..199c3cb60 100644 --- a/ppcls/configs/ImageNet/LeViT/LeViT_256.yaml +++ b/ppcls/configs/ImageNet/LeViT/LeViT_256.yaml @@ -17,7 +17,7 @@ Global: # model architecture Arch: - name: LeViT256 + name: LeViT_256 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/LeViT/LeViT_384.yaml b/ppcls/configs/ImageNet/LeViT/LeViT_384.yaml index fa53e1422..50f5f1b27 100644 --- a/ppcls/configs/ImageNet/LeViT/LeViT_384.yaml +++ b/ppcls/configs/ImageNet/LeViT/LeViT_384.yaml @@ -17,7 +17,7 @@ Global: # model architecture Arch: - name: LeViT384 + name: LeViT_384 # loss function config for traing/eval process Loss: