diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml index 6773af797..3f328c676 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,7 +16,8 @@ Global: # model architecture Arch: name: PPLCNet_x0_25 - + class_num: 1000 + # loss function config for traing/eval process Loss: Train: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_35.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_35.yaml index 36aa3dcb0..a3c01014f 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_35.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_35.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x0_35 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_5.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_5.yaml index 3ffc49385..8f5d2124b 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_5.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_5.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x0_5 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_75.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_75.yaml index 24ad8f373..b3fb75eeb 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_75.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_75.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x0_75 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0.yaml index 0921db540..62aff5a47 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_0.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x1_0 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_5.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_5.yaml index 290d1c0e6..390286dfc 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_5.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x1_5.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x1_5 + class_num: 1000 # loss function config for traing/eval process Loss: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_0.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_0.yaml index c8f7a3cdc..283e8afd9 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_0.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_0.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,7 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x2_0 - + class_num: 1000 # loss function config for traing/eval process Loss: Train: diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_5.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_5.yaml index 2ea69e30d..64efb3432 100644 --- a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_5.yaml +++ b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x2_5.yaml @@ -4,7 +4,6 @@ Global: pretrained_model: null output_dir: ./output/ device: gpu - class_num: 1000 save_interval: 1 eval_during_train: True eval_interval: 1 @@ -17,6 +16,7 @@ Global: # model architecture Arch: name: PPLCNet_x2_5 + class_num: 1000 # loss function config for traing/eval process Loss: