diff --git a/configs/AlexNet/AlexNet.yaml b/configs/AlexNet/AlexNet.yaml index 9ba5c6e9a..b9ef55aac 100644 --- a/configs/AlexNet/AlexNet.yaml +++ b/configs/AlexNet/AlexNet.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "AlexNet" +ARCHITECTURE: + name: "AlexNet" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DPN/DPN107.yaml b/configs/DPN/DPN107.yaml index 5da5cedec..b85a3678d 100644 --- a/configs/DPN/DPN107.yaml +++ b/configs/DPN/DPN107.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DPN107' +ARCHITECTURE: + name: 'DPN107' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DPN/DPN131.yaml b/configs/DPN/DPN131.yaml index a9ca71b22..b255ddd11 100644 --- a/configs/DPN/DPN131.yaml +++ b/configs/DPN/DPN131.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DPN131' +ARCHITECTURE: + name: 'DPN131' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DPN/DPN68.yaml b/configs/DPN/DPN68.yaml index f73045905..302545a65 100644 --- a/configs/DPN/DPN68.yaml +++ b/configs/DPN/DPN68.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DPN68' +ARCHITECTURE: + name: 'DPN68' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DPN/DPN92.yaml b/configs/DPN/DPN92.yaml index 078b45359..b081c06b6 100644 --- a/configs/DPN/DPN92.yaml +++ b/configs/DPN/DPN92.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DPN92' +ARCHITECTURE: + name: 'DPN92' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DPN/DPN98.yaml b/configs/DPN/DPN98.yaml index a577ffbcd..46e81e8e4 100644 --- a/configs/DPN/DPN98.yaml +++ b/configs/DPN/DPN98.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DPN98' +ARCHITECTURE: + name: 'DPN98' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DarkNet/DarkNet53.yaml b/configs/DarkNet/DarkNet53.yaml index 475491fe3..1b868c1e1 100644 --- a/configs/DarkNet/DarkNet53.yaml +++ b/configs/DarkNet/DarkNet53.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "DarkNet53" +ARCHITECTURE: + name: "DarkNet53" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DenseNet/DenseNet121.yaml b/configs/DenseNet/DenseNet121.yaml index 31af6194a..9c6532ed7 100644 --- a/configs/DenseNet/DenseNet121.yaml +++ b/configs/DenseNet/DenseNet121.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DenseNet121' +ARCHITECTURE: + name: 'DenseNet121' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DenseNet/DenseNet161.yaml b/configs/DenseNet/DenseNet161.yaml index c7fddce11..1976efedf 100644 --- a/configs/DenseNet/DenseNet161.yaml +++ b/configs/DenseNet/DenseNet161.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DenseNet161' +ARCHITECTURE: + name: 'DenseNet161' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DenseNet/DenseNet169.yaml b/configs/DenseNet/DenseNet169.yaml index 9b55b42cf..b5c429f19 100644 --- a/configs/DenseNet/DenseNet169.yaml +++ b/configs/DenseNet/DenseNet169.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DenseNet169' +ARCHITECTURE: + name: 'DenseNet169' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DenseNet/DenseNet201.yaml b/configs/DenseNet/DenseNet201.yaml index 20453bf3f..ce6935f83 100644 --- a/configs/DenseNet/DenseNet201.yaml +++ b/configs/DenseNet/DenseNet201.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DenseNet201' +ARCHITECTURE: + name: 'DenseNet201' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/DenseNet/DenseNet264.yaml b/configs/DenseNet/DenseNet264.yaml index 06f69c772..1cdc2092a 100644 --- a/configs/DenseNet/DenseNet264.yaml +++ b/configs/DenseNet/DenseNet264.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'DenseNet264' +ARCHITECTURE: + name: 'DenseNet264' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W18_C.yaml b/configs/HRNet/HRNet_W18_C.yaml index b71388d07..fd08f5e57 100644 --- a/configs/HRNet/HRNet_W18_C.yaml +++ b/configs/HRNet/HRNet_W18_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W18_C' +ARCHITECTURE: + name: 'HRNet_W18_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W30_C.yaml b/configs/HRNet/HRNet_W30_C.yaml index 2cd1b2ee0..6eaa2afb6 100644 --- a/configs/HRNet/HRNet_W30_C.yaml +++ b/configs/HRNet/HRNet_W30_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W30_C' +ARCHITECTURE: + name: 'HRNet_W30_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W32_C.yaml b/configs/HRNet/HRNet_W32_C.yaml index 5e27fc7ef..7dc926130 100644 --- a/configs/HRNet/HRNet_W32_C.yaml +++ b/configs/HRNet/HRNet_W32_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W32_C' +ARCHITECTURE: + name: 'HRNet_W32_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W40_C.yaml b/configs/HRNet/HRNet_W40_C.yaml index 5f9ad6e27..63b4088bd 100644 --- a/configs/HRNet/HRNet_W40_C.yaml +++ b/configs/HRNet/HRNet_W40_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W40_C' +ARCHITECTURE: + name: 'HRNet_W40_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W44_C.yaml b/configs/HRNet/HRNet_W44_C.yaml index 6c2f7cdc2..c3b91cb56 100644 --- a/configs/HRNet/HRNet_W44_C.yaml +++ b/configs/HRNet/HRNet_W44_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W44_C' +ARCHITECTURE: + name: 'HRNet_W44_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W48_C.yaml b/configs/HRNet/HRNet_W48_C.yaml index 978892617..0c253fdf6 100644 --- a/configs/HRNet/HRNet_W48_C.yaml +++ b/configs/HRNet/HRNet_W48_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W48_C' +ARCHITECTURE: + name: 'HRNet_W48_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/HRNet/HRNet_W64_C.yaml b/configs/HRNet/HRNet_W64_C.yaml index 3300bab06..e7f25b35c 100644 --- a/configs/HRNet/HRNet_W64_C.yaml +++ b/configs/HRNet/HRNet_W64_C.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'HRNet_W64_C' +ARCHITECTURE: + name: 'HRNet_W64_C' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Inception/GoogLeNet.yaml b/configs/Inception/GoogLeNet.yaml index 87789678a..c5469368e 100644 --- a/configs/Inception/GoogLeNet.yaml +++ b/configs/Inception/GoogLeNet.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "GoogLeNet" +ARCHITECTURE: + name: "GoogLeNet" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Inception/InceptionV4.yaml b/configs/Inception/InceptionV4.yaml index 42bcd0cd8..f418d335c 100644 --- a/configs/Inception/InceptionV4.yaml +++ b/configs/Inception/InceptionV4.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'InceptionV4' +ARCHITECTURE: + name: 'InceptionV4' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV1/MobileNetV1.yaml b/configs/MobileNetV1/MobileNetV1.yaml index 4f31699d2..e6db67e31 100644 --- a/configs/MobileNetV1/MobileNetV1.yaml +++ b/configs/MobileNetV1/MobileNetV1.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV1" +ARCHITECTURE: + name: "MobileNetV1" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV1/MobileNetV1_x0_25.yaml b/configs/MobileNetV1/MobileNetV1_x0_25.yaml index 6cd83247d..1f48a4812 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_25.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_25.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV1_x0_25" +ARCHITECTURE: + name: "MobileNetV1_x0_25" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV1/MobileNetV1_x0_5.yaml b/configs/MobileNetV1/MobileNetV1_x0_5.yaml index a54b0743c..9bf37f3aa 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_5.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV1_x0_5" +ARCHITECTURE: + name: "MobileNetV1_x0_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV1/MobileNetV1_x0_75.yaml b/configs/MobileNetV1/MobileNetV1_x0_75.yaml index d9841d850..cb6a6e4c5 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_75.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_75.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV1_x0_75" +ARCHITECTURE: + name: "MobileNetV1_x0_75" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2.yaml b/configs/MobileNetV2/MobileNetV2.yaml index 31895f705..11b07368f 100644 --- a/configs/MobileNetV2/MobileNetV2.yaml +++ b/configs/MobileNetV2/MobileNetV2.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2" +ARCHITECTURE: + name: "MobileNetV2" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2_x0_25.yaml b/configs/MobileNetV2/MobileNetV2_x0_25.yaml index a2b9ae894..cdcb7bf70 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_25.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_25.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2_x0_25" +ARCHITECTURE: + name: "MobileNetV2_x0_25" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2_x0_5.yaml b/configs/MobileNetV2/MobileNetV2_x0_5.yaml index 9361c4ce3..dba62d20d 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_5.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2_x0_5" +ARCHITECTURE: + name: "MobileNetV2_x0_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2_x0_75.yaml b/configs/MobileNetV2/MobileNetV2_x0_75.yaml index 75ca6f32c..92ccd0a4e 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_75.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_75.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2_x0_75" +ARCHITECTURE: + name: "MobileNetV2_x0_75" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2_x1_5.yaml b/configs/MobileNetV2/MobileNetV2_x1_5.yaml index 3eeaae742..9999aa9bb 100644 --- a/configs/MobileNetV2/MobileNetV2_x1_5.yaml +++ b/configs/MobileNetV2/MobileNetV2_x1_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2_x1_5" +ARCHITECTURE: + name: "MobileNetV2_x1_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV2/MobileNetV2_x2_0.yaml b/configs/MobileNetV2/MobileNetV2_x2_0.yaml index cc5358c1e..93f2ee67a 100644 --- a/configs/MobileNetV2/MobileNetV2_x2_0.yaml +++ b/configs/MobileNetV2/MobileNetV2_x2_0.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV2_x2_0" +ARCHITECTURE: + name: "MobileNetV2_x2_0" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml index 32d168c80..2f42c00b5 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_large_x0_35" +ARCHITECTURE: + name: "MobileNetV3_large_x0_35" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml index b3580040d..34389fef0 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_large_x0_5" +ARCHITECTURE: + name: "MobileNetV3_large_x0_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml index e4bcfbac6..4666c66b1 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_large_x0_75" +ARCHITECTURE: + name: "MobileNetV3_large_x0_75" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml index 0477d1a3e..9c4856b27 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_large_x1_0" +ARCHITECTURE: + name: "MobileNetV3_large_x1_0" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml index 00670ad1d..2a1840b9f 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_large_x1_25" +ARCHITECTURE: + name: "MobileNetV3_large_x1_25" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml index afec8763e..dca65b7d3 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_small_x0_35" +ARCHITECTURE: + name: "MobileNetV3_small_x0_35" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml index 9c7d3e334..36fd90f40 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_small_x0_5" +ARCHITECTURE: + name: "MobileNetV3_small_x0_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml index 4febf84dc..bae9e83a1 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_small_x0_75" +ARCHITECTURE: + name: "MobileNetV3_small_x0_75" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml index da4690484..141c42cf4 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_small_x1_0" +ARCHITECTURE: + name: "MobileNetV3_small_x1_0" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml index c8fd74675..e279ffb28 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "MobileNetV3_small_x1_25" +ARCHITECTURE: + name: "MobileNetV3_small_x1_25" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Res2Net/Res2Net101_vd_26w_4s.yaml b/configs/Res2Net/Res2Net101_vd_26w_4s.yaml index 497f2bfa9..af6ea7522 100644 --- a/configs/Res2Net/Res2Net101_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net101_vd_26w_4s.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'Res2Net101_vd_26w_4s' +ARCHITECTURE: + name: 'Res2Net101_vd_26w_4s' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Res2Net/Res2Net200_vd_26w_4s.yaml b/configs/Res2Net/Res2Net200_vd_26w_4s.yaml index cd57c30bb..3c610271c 100644 --- a/configs/Res2Net/Res2Net200_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net200_vd_26w_4s.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'Res2Net200_vd_26w_4s' +ARCHITECTURE: + name: 'Res2Net200_vd_26w_4s' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Res2Net/Res2Net50_14w_8s.yaml b/configs/Res2Net/Res2Net50_14w_8s.yaml index 98e823424..28ddef49f 100644 --- a/configs/Res2Net/Res2Net50_14w_8s.yaml +++ b/configs/Res2Net/Res2Net50_14w_8s.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'Res2Net50_14w_8s' +ARCHITECTURE: + name: 'Res2Net50_14w_8s' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Res2Net/Res2Net50_26w_4s.yaml b/configs/Res2Net/Res2Net50_26w_4s.yaml index 8eace214b..16b6ba4f0 100644 --- a/configs/Res2Net/Res2Net50_26w_4s.yaml +++ b/configs/Res2Net/Res2Net50_26w_4s.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'Res2Net50_26w_4s' +ARCHITECTURE: + name: 'Res2Net50_26w_4s' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/Res2Net/Res2Net50_vd_26w_4s.yaml b/configs/Res2Net/Res2Net50_vd_26w_4s.yaml index aac1dcaec..51c865f29 100644 --- a/configs/Res2Net/Res2Net50_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net50_vd_26w_4s.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'Res2Net50_vd_26w_4s' +ARCHITECTURE: + name: 'Res2Net50_vd_26w_4s' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt101_32x4d.yaml b/configs/ResNeXt/ResNeXt101_32x4d.yaml index 4088cc8c2..80dcec7cd 100644 --- a/configs/ResNeXt/ResNeXt101_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt101_32x4d' +ARCHITECTURE: + name: 'ResNeXt101_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt101_64x4d.yaml b/configs/ResNeXt/ResNeXt101_64x4d.yaml index 091d04aea..4b6b6ade0 100644 --- a/configs/ResNeXt/ResNeXt101_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt101_64x4d' +ARCHITECTURE: + name: 'ResNeXt101_64x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml index 40b331464..0770578c1 100644 --- a/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt101_vd_32x4d' +ARCHITECTURE: + name: 'ResNeXt101_vd_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml index b9cc0b95e..9a61f80e2 100644 --- a/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt101_vd_64x4d' +ARCHITECTURE: + name: 'ResNeXt101_vd_64x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt152_32x4d.yaml b/configs/ResNeXt/ResNeXt152_32x4d.yaml index 287013d27..5c79efd65 100644 --- a/configs/ResNeXt/ResNeXt152_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt152_32x4d' +ARCHITECTURE: + name: 'ResNeXt152_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt152_64x4d.yaml b/configs/ResNeXt/ResNeXt152_64x4d.yaml index 27f0919cd..f129c43e3 100644 --- a/configs/ResNeXt/ResNeXt152_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt152_64x4d' +ARCHITECTURE: + name: 'ResNeXt152_64x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml index 341faca16..62a44ef4a 100644 --- a/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt152_vd_32x4d' +ARCHITECTURE: + name: 'ResNeXt152_vd_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml index ea37b955a..a204c9442 100644 --- a/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt152_vd_64x4d' +ARCHITECTURE: + name: 'ResNeXt152_vd_64x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt50_32x4d.yaml b/configs/ResNeXt/ResNeXt50_32x4d.yaml index 21efc0ad0..b4b551d6f 100644 --- a/configs/ResNeXt/ResNeXt50_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt50_32x4d' +ARCHITECTURE: + name: 'ResNeXt50_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt50_64x4d.yaml b/configs/ResNeXt/ResNeXt50_64x4d.yaml index 2887a3425..216ae6962 100644 --- a/configs/ResNeXt/ResNeXt50_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ResNeXt50_64x4d" +ARCHITECTURE: + name: "ResNeXt50_64x4d" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml index 1af4f5282..c78c72df1 100644 --- a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ResNeXt50_vd_32x4d" +ARCHITECTURE: + name: "ResNeXt50_vd_32x4d" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml index ea17291a8..8441a020c 100644 --- a/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNeXt50_vd_64x4d' +ARCHITECTURE: + name: 'ResNeXt50_vd_64x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet101.yaml b/configs/ResNet/ResNet101.yaml index 5f2f398a0..cc1001a5e 100644 --- a/configs/ResNet/ResNet101.yaml +++ b/configs/ResNet/ResNet101.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet101' +ARCHITECTURE: + name: 'ResNet101' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet101_vd.yaml b/configs/ResNet/ResNet101_vd.yaml index c29df8da1..e137e296e 100644 --- a/configs/ResNet/ResNet101_vd.yaml +++ b/configs/ResNet/ResNet101_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet101_vd' +ARCHITECTURE: + name: 'ResNet101_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet152.yaml b/configs/ResNet/ResNet152.yaml index d1918b300..932efdfd5 100644 --- a/configs/ResNet/ResNet152.yaml +++ b/configs/ResNet/ResNet152.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet152' +ARCHITECTURE: + name: 'ResNet152' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet152_vd.yaml b/configs/ResNet/ResNet152_vd.yaml index b200663d2..5fe235515 100644 --- a/configs/ResNet/ResNet152_vd.yaml +++ b/configs/ResNet/ResNet152_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet152_vd' +ARCHITECTURE: + name: 'ResNet152_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet18.yaml b/configs/ResNet/ResNet18.yaml index 033be21a1..c362ff742 100644 --- a/configs/ResNet/ResNet18.yaml +++ b/configs/ResNet/ResNet18.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet18' +ARCHITECTURE: + name: 'ResNet18' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet18_vd.yaml b/configs/ResNet/ResNet18_vd.yaml index 690ce3337..fd9b406d0 100644 --- a/configs/ResNet/ResNet18_vd.yaml +++ b/configs/ResNet/ResNet18_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet18_vd' +ARCHITECTURE: + name: 'ResNet18_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet200_vd.yaml b/configs/ResNet/ResNet200_vd.yaml index 691ca702f..d292fba0a 100644 --- a/configs/ResNet/ResNet200_vd.yaml +++ b/configs/ResNet/ResNet200_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet200_vd' +ARCHITECTURE: + name: 'ResNet200_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet34.yaml b/configs/ResNet/ResNet34.yaml index 2ca0c702f..19a363a65 100644 --- a/configs/ResNet/ResNet34.yaml +++ b/configs/ResNet/ResNet34.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet34' +ARCHITECTURE: + name: 'ResNet34' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet34_vd.yaml b/configs/ResNet/ResNet34_vd.yaml index 5b086d428..811b321bd 100644 --- a/configs/ResNet/ResNet34_vd.yaml +++ b/configs/ResNet/ResNet34_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet34_vd' +ARCHITECTURE: + name: 'ResNet34_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet50.yaml b/configs/ResNet/ResNet50.yaml index 8150e05f0..b8b0abd82 100644 --- a/configs/ResNet/ResNet50.yaml +++ b/configs/ResNet/ResNet50.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet50' +ARCHITECTURE: + name: 'ResNet50' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet50_vc.yaml b/configs/ResNet/ResNet50_vc.yaml index 59cd1a220..3680fc7d2 100644 --- a/configs/ResNet/ResNet50_vc.yaml +++ b/configs/ResNet/ResNet50_vc.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet50_vc' +ARCHITECTURE: + name: 'ResNet50_vc' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet/ResNet50_vd.yaml b/configs/ResNet/ResNet50_vd.yaml index b0b68b979..28696c6d4 100644 --- a/configs/ResNet/ResNet50_vd.yaml +++ b/configs/ResNet/ResNet50_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'ResNet50_vd' +ARCHITECTURE: + name: 'ResNet50_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ResNet_ACNet/ResNet_ACNet.yaml b/configs/ResNet_ACNet/ResNet_ACNet.yaml index fa7fcf802..b5d4200c7 100644 --- a/configs/ResNet_ACNet/ResNet_ACNet.yaml +++ b/configs/ResNet_ACNet/ResNet_ACNet.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ResNet_ACNet" +ARCHITECTURE: + name: "ResNet_ACNet" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SENet154_vd.yaml b/configs/SENet/SENet154_vd.yaml index f60595719..7a105f378 100644 --- a/configs/SENet/SENet154_vd.yaml +++ b/configs/SENet/SENet154_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SENet154_vd' +ARCHITECTURE: + name: 'SENet154_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNeXt101_32x4d.yaml b/configs/SENet/SE_ResNeXt101_32x4d.yaml index 1a9887b53..4b855b52b 100644 --- a/configs/SENet/SE_ResNeXt101_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt101_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNeXt101_32x4d' +ARCHITECTURE: + name: 'SE_ResNeXt101_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNeXt50_32x4d.yaml b/configs/SENet/SE_ResNeXt50_32x4d.yaml index 1200d87b8..4625f163c 100644 --- a/configs/SENet/SE_ResNeXt50_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt50_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNeXt50_32x4d' +ARCHITECTURE: + name: 'SE_ResNeXt50_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml b/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml index 35f25500c..2584901f3 100644 --- a/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNeXt50_vd_32x4d' +ARCHITECTURE: + name: 'SE_ResNeXt50_vd_32x4d' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNet18_vd.yaml b/configs/SENet/SE_ResNet18_vd.yaml index 007a179df..4db7506ec 100644 --- a/configs/SENet/SE_ResNet18_vd.yaml +++ b/configs/SENet/SE_ResNet18_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNet18_vd' +ARCHITECTURE: + name: 'SE_ResNet18_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNet34_vd.yaml b/configs/SENet/SE_ResNet34_vd.yaml index 3ba4a804b..942a7fefc 100644 --- a/configs/SENet/SE_ResNet34_vd.yaml +++ b/configs/SENet/SE_ResNet34_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNet34_vd' +ARCHITECTURE: + name: 'SE_ResNet34_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SENet/SE_ResNet50_vd.yaml b/configs/SENet/SE_ResNet50_vd.yaml index 497a7f37b..177ff8970 100644 --- a/configs/SENet/SE_ResNet50_vd.yaml +++ b/configs/SENet/SE_ResNet50_vd.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: 'SE_ResNet50_vd' +ARCHITECTURE: + name: 'SE_ResNet50_vd' + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2.yaml b/configs/ShuffleNet/ShuffleNetV2.yaml index 5f4b4e83f..c097afaea 100644 --- a/configs/ShuffleNet/ShuffleNetV2.yaml +++ b/configs/ShuffleNet/ShuffleNetV2.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2" +ARCHITECTURE: + name: "ShuffleNetV2" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_swish.yaml b/configs/ShuffleNet/ShuffleNetV2_swish.yaml index e3e55a6ed..4e64ce8bb 100644 --- a/configs/ShuffleNet/ShuffleNetV2_swish.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_swish.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_swish" +ARCHITECTURE: + name: "ShuffleNetV2_swish" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml index f80640b55..996f040bb 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_x0_25" +ARCHITECTURE: + name: "ShuffleNetV2_x0_25" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml index fe13a98d6..f29414741 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_x0_33" +ARCHITECTURE: + name: "ShuffleNetV2_x0_33" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml index 85f930603..05a1ad3eb 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_x0_5" +ARCHITECTURE: + name: "ShuffleNetV2_x0_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml index 3fb7dac9b..63f50d484 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_x1_5" +ARCHITECTURE: + name: "ShuffleNetV2_x1_5" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml index a68d79517..5a14cebbf 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "ShuffleNetV2_x2_0" +ARCHITECTURE: + name: "ShuffleNetV2_x2_0" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SqueezeNet/SqueezeNet1_0.yaml b/configs/SqueezeNet/SqueezeNet1_0.yaml index 4edb704e0..9107bdfe1 100644 --- a/configs/SqueezeNet/SqueezeNet1_0.yaml +++ b/configs/SqueezeNet/SqueezeNet1_0.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "SqueezeNet1_0" +ARCHITECTURE: + name: "SqueezeNet1_0" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/SqueezeNet/SqueezeNet1_1.yaml b/configs/SqueezeNet/SqueezeNet1_1.yaml index 891d88a2b..f32b55980 100644 --- a/configs/SqueezeNet/SqueezeNet1_1.yaml +++ b/configs/SqueezeNet/SqueezeNet1_1.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "SqueezeNet1_1" +ARCHITECTURE: + name: "SqueezeNet1_1" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/VGG/VGG11.yaml b/configs/VGG/VGG11.yaml index f1d8ab34c..9331f52f9 100644 --- a/configs/VGG/VGG11.yaml +++ b/configs/VGG/VGG11.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "VGG11" +ARCHITECTURE: + name: "VGG11" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/VGG/VGG13.yaml b/configs/VGG/VGG13.yaml index cae3b7d58..67420fe13 100644 --- a/configs/VGG/VGG13.yaml +++ b/configs/VGG/VGG13.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "VGG13" +ARCHITECTURE: + name: "VGG13" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/VGG/VGG16.yaml b/configs/VGG/VGG16.yaml index 26f451e9b..0e6648062 100644 --- a/configs/VGG/VGG16.yaml +++ b/configs/VGG/VGG16.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "VGG16" +ARCHITECTURE: + name: "VGG16" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/VGG/VGG19.yaml b/configs/VGG/VGG19.yaml index 21da446ff..3e4299791 100644 --- a/configs/VGG/VGG19.yaml +++ b/configs/VGG/VGG19.yaml @@ -1,5 +1,7 @@ mode: 'train' -architecture: "VGG19" +ARCHITECTURE: + name: "VGG19" + pretrained_model: "" model_save_dir: "./output/" classes_num: 1000 diff --git a/configs/eval.yaml b/configs/eval.yaml index d9404f98b..6c5333cb0 100644 --- a/configs/eval.yaml +++ b/configs/eval.yaml @@ -1,5 +1,7 @@ mode: 'valid' -architecture: "" +ARCHITECTURE: + name: "" + pretrained_model: "" classes_num: 1000 total_images: 1281167