diff --git a/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml b/ppcls/configs/ImageNet/PPLCNet/PPLCNet_x0_25.yaml index 6773af797e05013eea1bf676876e91ebc1322fee..3f328c67676c7983e7e46c24947fb042e24fccb7 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 36aa3dcb0b269b456550aa8dce6cb94866661303..a3c01014f5c9abdf31c890ff912e089bf30d0fba 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 3ffc493851076f8ddd692585de966647a5bc3f57..8f5d2124bcf423a2e74a43ba37aacd8ba85d7791 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 24ad8f3733b2b4fca7f43fc142f539c4d6d31520..b3fb75eeb03f68003b3bb5dea46525b59f315525 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 0921db540c26caf075d57cb520397a08c932f49b..62aff5a474dbd12bda855ecfd7615c4870145c5c 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 290d1c0e69812d43d76167c48ee6d5357baa2b96..390286dfc776f86fb268f012002d8eac3df0b5ae 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 c8f7a3cdc40c0d6a307c30b2167588e02ea892d7..283e8afd9e8840dc61444a1d432e1c1dcb42d08f 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 2ea69e30d0a45385cb6113418c4fc244055de2ce..64efb34326d1ac8d44bd70111c83411712cb2fb9 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: