From dade4ef46b1d926f7cc4dbd4c35963940ab15cd0 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Tue, 29 Jun 2021 20:28:55 +0800 Subject: [PATCH] fix class num (#989) --- ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml | 2 +- ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml | 2 +- ppcls/configs/ImageNet/ResNet/ResNet101.yaml | 2 +- ppcls/configs/ImageNet/ResNet/ResNet152.yaml | 2 +- ppcls/configs/ImageNet/ResNet/ResNet50.yaml | 2 +- .../SwinTransformer_base_patch4_window12_384.yaml | 2 +- .../SwinTransformer_base_patch4_window7_224.yaml | 2 +- .../SwinTransformer_large_patch4_window12_384.yaml | 2 +- .../SwinTransformer_large_patch4_window7_224.yaml | 2 +- .../SwinTransformer_small_patch4_window7_224.yaml | 2 +- .../SwinTransformer_tiny_patch4_window7_224.yaml | 2 +- 11 files changed, 11 insertions(+), 11 deletions(-) diff --git a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml index 191c3dd1a..a39496d7b 100644 --- a/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml +++ b/ppcls/configs/ImageNet/MobileNetV1/MobileNetV1.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: MobileNetV1 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml index d37f2c962..40fc2d1f4 100644 --- a/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml +++ b/ppcls/configs/ImageNet/MobileNetV2/MobileNetV2.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: MobileNetV2 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/ResNet/ResNet101.yaml b/ppcls/configs/ImageNet/ResNet/ResNet101.yaml index 3df85629a..a4f228f26 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet101.yaml +++ b/ppcls/configs/ImageNet/ResNet/ResNet101.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: ResNet101 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/ResNet/ResNet152.yaml b/ppcls/configs/ImageNet/ResNet/ResNet152.yaml index 426160584..b91571322 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet152.yaml +++ b/ppcls/configs/ImageNet/ResNet/ResNet152.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: ResNet152 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/ResNet/ResNet50.yaml b/ppcls/configs/ImageNet/ResNet/ResNet50.yaml index 735a74e7d..b338b6395 100644 --- a/ppcls/configs/ImageNet/ResNet/ResNet50.yaml +++ b/ppcls/configs/ImageNet/ResNet/ResNet50.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: ResNet50 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml index 443371c6d..fa48840c8 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window12_384.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_base_patch4_window12_384 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml index 1c940d781..aa05383dd 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_base_patch4_window7_224.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_base_patch4_window7_224 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml index 2a0e12d49..c4eeaa2c6 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window12_384.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_large_patch4_window12_384 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml index 24b677bb8..e6bfc460f 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_large_patch4_window7_224.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_large_patch4_window7_224 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml index d402d33a7..f3bcad069 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_small_patch4_window7_224.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_small_patch4_window7_224 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml index b721f632a..390db2be0 100644 --- a/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml +++ b/ppcls/configs/ImageNet/SwinTransformer/SwinTransformer_tiny_patch4_window7_224.yaml @@ -17,9 +17,9 @@ Global: to_static: False # model architecture - class_num: 1000 Arch: name: SwinTransformer_tiny_patch4_window7_224 + class_num: 1000 # loss function config for traing/eval process Loss: