Skip to content

Commit 69cbcab

Browse files
committed
unify class_num to num_classes in model
1 parent d0ba764 commit 69cbcab

22 files changed

+22
-22
lines changed

tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_1n8c_dp_fp32.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ EMA:
3737
Model:
3838
name: convnext_base
3939
drop_path_rate : 0.5
40-
class_num: 1000
40+
num_classes: 1000
4141

4242
# loss function config for traing/eval process
4343
Loss:

tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ EMA:
3838
Model:
3939
name: convnext_base
4040
drop_path_rate : 0.5
41-
class_num: 1000
41+
num_classes: 1000
4242

4343
# loss function config for traing/eval process
4444
Loss:

tasks/classification/convnext/configs/ConvNeXt_base_224_in1k_4n32c_dp_fp32.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ EMA:
3737
Model:
3838
name: convnext_base
3939
drop_path_rate : 0.5
40-
class_num: 1000
40+
num_classes: 1000
4141

4242
# loss function config for traing/eval process
4343
Loss:

tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Model:
3232
name: DeiT_base_patch16_224
3333
drop_path_rate : 0.1
3434
drop_rate : 0.0
35-
class_num: 1000
35+
num_classes: 1000
3636

3737
# loss function config for traing/eval process
3838
Loss:

tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_1n8c_dp_fp32.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Model:
3232
name: DeiT_base_patch16_224
3333
drop_path_rate : 0.1
3434
drop_rate : 0.0
35-
class_num: 1000
35+
num_classes: 1000
3636

3737
# loss function config for traing/eval process
3838
Loss:

tasks/classification/deit/configs/DeiT_base_patch16_224_in1k_2n16c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ Model:
3232
name: DeiT_base_patch16_224
3333
drop_path_rate : 0.1
3434
drop_rate : 0.0
35-
class_num: 1000
35+
num_classes: 1000
3636

3737
# loss function config for traing/eval process
3838
Loss:

tasks/classification/vit/configs/ViT_base_patch16_224_in1k_1n8c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ DistributedStrategy:
3030
# model architecture
3131
Model:
3232
name: ViT_base_patch16_224
33-
class_num: 1000
33+
num_classes: 1000
3434
drop_rate: 0.1
3535

3636
# loss function config for traing/eval process

tasks/classification/vit/configs/ViT_base_patch16_384_ft_in1k_1n8c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ DistributedStrategy:
3131
# model architecture
3232
Model:
3333
name: ViT_base_patch16_384
34-
class_num: 1000
34+
num_classes: 1000
3535
drop_rate: 0.1
3636

3737
# loss function config for traing/eval process

tasks/classification/vit/configs/ViT_large_patch16_224_in21k_4n32c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ DistributedStrategy:
3030
# model architecture
3131
Model:
3232
name: ViT_large_patch16_224
33-
class_num: 21841
33+
num_classes: 21841
3434
drop_rate: 0.1
3535

3636
# loss function config for traing/eval process

tasks/classification/vit/configs/ViT_large_patch16_384_in1k_ft_4n32c_dp_fp16o2.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ DistributedStrategy:
3131
# model architecture
3232
Model:
3333
name: ViT_large_patch16_384
34-
class_num: 1000
34+
num_classes: 1000
3535
drop_rate: 0.1
3636

3737
# loss function config for traing/eval process

0 commit comments

Comments
 (0)