unify class_num to num_classes

TingquanGao · TingquanGao · commit d0ba7644f589 · 2023-06-28T03:36:57.000Z
diff --git a/configs/deit/deit-base-p16-pt_in1k-224_2n16c_fp16_o1_dp.yaml b/configs/deit/deit-base-p16-pt_in1k-224_2n16c_fp16_o1_dp.yaml
@@ -27,7 +27,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.0
       drop_path_rate : 0.1
 
diff --git a/configs/lvvit/lvvit_tiny.yaml b/configs/lvvit/lvvit_tiny.yaml
@@ -30,7 +30,7 @@ model:
       skip_lam: 1
       return_dense: True
       mix_token: True
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.0
       drop_path_rate : 0.1
 
diff --git a/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_dp.yaml b/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_dp.yaml
@@ -27,7 +27,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
diff --git a/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_sharding.yaml b/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o1_sharding.yaml
@@ -33,7 +33,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
diff --git a/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_dp.yaml b/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_dp.yaml
@@ -27,7 +27,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
diff --git a/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_sharding.yaml b/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp16_o2_sharding.yaml
@@ -33,7 +33,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
diff --git a/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp32_dp.yaml b/configs/vision_transformer/vit-base-p16-pt_in1k-224_4n32c_fp32_dp.yaml
@@ -15,7 +15,7 @@ model:
       mlp_ratio: 4
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
diff --git a/configs/vision_transformer/vit-g-p14-pt_in1k-224_1n8c.yaml b/configs/vision_transformer/vit-g-p14-pt_in1k-224_1n8c.yaml
@@ -15,7 +15,7 @@ model:
       mlp_ratio: 4.9231
       qkv_bias: True
       epsilon: 1e-6
-      class_num: 1000
+      num_classes: 1000
       drop_rate: 0.1
       representation_size: 768
   label_smoothing: 0.0001
