From d70680deda1b9b793a4909be7012660b6e597540 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 00:11:03 +0800 Subject: [PATCH 01/14] change model_save_dir --- configs/AlexNet/AlexNet.yaml | 5 +---- configs/DPN/DPN107.yaml | 2 +- configs/DPN/DPN131.yaml | 2 +- configs/DPN/DPN68.yaml | 2 +- configs/DPN/DPN92.yaml | 2 +- configs/DPN/DPN98.yaml | 2 +- configs/DarkNet/DarkNet53.yaml | 2 +- configs/DenseNet/DenseNet121.yaml | 2 +- configs/DenseNet/DenseNet161.yaml | 2 +- configs/DenseNet/DenseNet169.yaml | 2 +- configs/DenseNet/DenseNet201.yaml | 2 +- configs/DenseNet/DenseNet264.yaml | 2 +- configs/HRNet/HRNet_W18_C.yaml | 2 +- configs/HRNet/HRNet_W30_C.yaml | 2 +- configs/HRNet/HRNet_W32_C.yaml | 2 +- configs/HRNet/HRNet_W40_C.yaml | 2 +- configs/HRNet/HRNet_W44_C.yaml | 2 +- configs/HRNet/HRNet_W48_C.yaml | 2 +- configs/HRNet/HRNet_W64_C.yaml | 2 +- configs/Inception/GoogLeNet.yaml | 2 +- configs/Inception/InceptionV4.yaml | 2 +- configs/MobileNetV1/MobileNetV1.yaml | 2 +- configs/MobileNetV1/MobileNetV1_x0_25.yaml | 2 +- configs/MobileNetV1/MobileNetV1_x0_5.yaml | 2 +- configs/MobileNetV1/MobileNetV1_x0_75.yaml | 2 +- configs/MobileNetV2/MobileNetV2.yaml | 2 +- configs/MobileNetV2/MobileNetV2_x0_25.yaml | 2 +- configs/MobileNetV2/MobileNetV2_x0_5.yaml | 2 +- configs/MobileNetV2/MobileNetV2_x0_75.yaml | 2 +- configs/MobileNetV2/MobileNetV2_x1_5.yaml | 2 +- configs/MobileNetV2/MobileNetV2_x2_0.yaml | 2 +- configs/MobileNetV3/MobileNetV3_large_x0_35.yaml | 2 +- configs/MobileNetV3/MobileNetV3_large_x0_5.yaml | 2 +- configs/MobileNetV3/MobileNetV3_large_x0_75.yaml | 2 +- configs/MobileNetV3/MobileNetV3_large_x1_0.yaml | 2 +- configs/MobileNetV3/MobileNetV3_large_x1_25.yaml | 2 +- configs/MobileNetV3/MobileNetV3_small_x0_35.yaml | 2 +- configs/MobileNetV3/MobileNetV3_small_x0_5.yaml | 2 +- configs/MobileNetV3/MobileNetV3_small_x0_75.yaml | 2 +- configs/MobileNetV3/MobileNetV3_small_x1_0.yaml | 2 +- configs/MobileNetV3/MobileNetV3_small_x1_25.yaml | 2 +- configs/Res2Net/Res2Net101_vd_26w_4s.yaml | 2 +- configs/Res2Net/Res2Net200_vd_26w_4s.yaml | 2 +- configs/Res2Net/Res2Net50_14w_8s.yaml | 2 +- configs/Res2Net/Res2Net50_26w_4s.yaml | 2 +- configs/Res2Net/Res2Net50_vd_26w_4s.yaml | 2 +- configs/ResNeXt/ResNeXt101_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt101_64x4d.yaml | 2 +- configs/ResNeXt/ResNeXt101_vd_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt101_vd_64x4d.yaml | 2 +- configs/ResNeXt/ResNeXt152_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt152_64x4d.yaml | 2 +- configs/ResNeXt/ResNeXt152_vd_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt152_vd_64x4d.yaml | 2 +- configs/ResNeXt/ResNeXt50_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt50_64x4d.yaml | 2 +- configs/ResNeXt/ResNeXt50_vd_32x4d.yaml | 2 +- configs/ResNeXt/ResNeXt50_vd_64x4d.yaml | 2 +- configs/ResNet/ResNet101.yaml | 2 +- configs/ResNet/ResNet101_vd.yaml | 2 +- configs/ResNet/ResNet152.yaml | 2 +- configs/ResNet/ResNet152_vd.yaml | 2 +- configs/ResNet/ResNet18.yaml | 2 +- configs/ResNet/ResNet18_vd.yaml | 2 +- configs/ResNet/ResNet200_vd.yaml | 2 +- configs/ResNet/ResNet34.yaml | 2 +- configs/ResNet/ResNet34_vd.yaml | 2 +- configs/ResNet/ResNet50.yaml | 2 +- configs/ResNet/ResNet50_vc.yaml | 2 +- configs/ResNet/ResNet50_vd.yaml | 2 +- configs/ResNet_ACNet/ResNet_ACNet.yaml | 2 +- configs/SENet/SENet154_vd.yaml | 2 +- configs/SENet/SE_ResNeXt101_32x4d.yaml | 2 +- configs/SENet/SE_ResNeXt50_32x4d.yaml | 2 +- configs/SENet/SE_ResNeXt50_vd_32x4d.yaml | 2 +- configs/SENet/SE_ResNet18_vd.yaml | 2 +- configs/SENet/SE_ResNet34_vd.yaml | 2 +- configs/SENet/SE_ResNet50_vd.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_swish.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_x0_25.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_x0_33.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_x0_5.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_x1_5.yaml | 2 +- configs/ShuffleNet/ShuffleNetV2_x2_0.yaml | 2 +- configs/SqueezeNet/SqueezeNet1_0.yaml | 2 +- configs/SqueezeNet/SqueezeNet1_1.yaml | 2 +- configs/VGG/VGG11.yaml | 2 +- configs/VGG/VGG13.yaml | 2 +- configs/VGG/VGG16.yaml | 2 +- configs/VGG/VGG19.yaml | 2 +- configs/eval.yaml | 1 - 92 files changed, 91 insertions(+), 95 deletions(-) diff --git a/configs/AlexNet/AlexNet.yaml b/configs/AlexNet/AlexNet.yaml index fa46e6804..9ba5c6e9a 100644 --- a/configs/AlexNet/AlexNet.yaml +++ b/configs/AlexNet/AlexNet.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "AlexNet" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 @@ -48,8 +48,6 @@ TRAIN: order: '' - ToCHWImage: - - VALID: batch_size: 64 num_workers: 4 @@ -72,4 +70,3 @@ VALID: order: '' - ToCHWImage: - diff --git a/configs/DPN/DPN107.yaml b/configs/DPN/DPN107.yaml index d44418fe7..5da5cedec 100644 --- a/configs/DPN/DPN107.yaml +++ b/configs/DPN/DPN107.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DPN107' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DPN/DPN131.yaml b/configs/DPN/DPN131.yaml index 95b6345a7..a9ca71b22 100644 --- a/configs/DPN/DPN131.yaml +++ b/configs/DPN/DPN131.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DPN131' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DPN/DPN68.yaml b/configs/DPN/DPN68.yaml index e1fa30b43..f73045905 100644 --- a/configs/DPN/DPN68.yaml +++ b/configs/DPN/DPN68.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DPN68' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DPN/DPN92.yaml b/configs/DPN/DPN92.yaml index ecfae64dc..078b45359 100644 --- a/configs/DPN/DPN92.yaml +++ b/configs/DPN/DPN92.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DPN92' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DPN/DPN98.yaml b/configs/DPN/DPN98.yaml index 51dab0ac6..a577ffbcd 100644 --- a/configs/DPN/DPN98.yaml +++ b/configs/DPN/DPN98.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DPN98' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DarkNet/DarkNet53.yaml b/configs/DarkNet/DarkNet53.yaml index e3610d9a4..475491fe3 100644 --- a/configs/DarkNet/DarkNet53.yaml +++ b/configs/DarkNet/DarkNet53.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "DarkNet53" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DenseNet/DenseNet121.yaml b/configs/DenseNet/DenseNet121.yaml index 2b13cacb7..31af6194a 100644 --- a/configs/DenseNet/DenseNet121.yaml +++ b/configs/DenseNet/DenseNet121.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DenseNet121' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DenseNet/DenseNet161.yaml b/configs/DenseNet/DenseNet161.yaml index c69376126..c7fddce11 100644 --- a/configs/DenseNet/DenseNet161.yaml +++ b/configs/DenseNet/DenseNet161.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DenseNet161' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DenseNet/DenseNet169.yaml b/configs/DenseNet/DenseNet169.yaml index d8f80309b..9b55b42cf 100644 --- a/configs/DenseNet/DenseNet169.yaml +++ b/configs/DenseNet/DenseNet169.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DenseNet169' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DenseNet/DenseNet201.yaml b/configs/DenseNet/DenseNet201.yaml index c848de2dd..20453bf3f 100644 --- a/configs/DenseNet/DenseNet201.yaml +++ b/configs/DenseNet/DenseNet201.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DenseNet201' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/DenseNet/DenseNet264.yaml b/configs/DenseNet/DenseNet264.yaml index 4794be4ee..06f69c772 100644 --- a/configs/DenseNet/DenseNet264.yaml +++ b/configs/DenseNet/DenseNet264.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'DenseNet264' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W18_C.yaml b/configs/HRNet/HRNet_W18_C.yaml index d74e16318..b71388d07 100644 --- a/configs/HRNet/HRNet_W18_C.yaml +++ b/configs/HRNet/HRNet_W18_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W18_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W30_C.yaml b/configs/HRNet/HRNet_W30_C.yaml index c270a7f49..2cd1b2ee0 100644 --- a/configs/HRNet/HRNet_W30_C.yaml +++ b/configs/HRNet/HRNet_W30_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W30_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W32_C.yaml b/configs/HRNet/HRNet_W32_C.yaml index 97f748f01..5e27fc7ef 100644 --- a/configs/HRNet/HRNet_W32_C.yaml +++ b/configs/HRNet/HRNet_W32_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W32_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W40_C.yaml b/configs/HRNet/HRNet_W40_C.yaml index cf6d3e81e..5f9ad6e27 100644 --- a/configs/HRNet/HRNet_W40_C.yaml +++ b/configs/HRNet/HRNet_W40_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W40_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W44_C.yaml b/configs/HRNet/HRNet_W44_C.yaml index 2e435a635..6c2f7cdc2 100644 --- a/configs/HRNet/HRNet_W44_C.yaml +++ b/configs/HRNet/HRNet_W44_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W44_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W48_C.yaml b/configs/HRNet/HRNet_W48_C.yaml index b63341d7c..978892617 100644 --- a/configs/HRNet/HRNet_W48_C.yaml +++ b/configs/HRNet/HRNet_W48_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W48_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/HRNet/HRNet_W64_C.yaml b/configs/HRNet/HRNet_W64_C.yaml index 8684664ab..3300bab06 100644 --- a/configs/HRNet/HRNet_W64_C.yaml +++ b/configs/HRNet/HRNet_W64_C.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'HRNet_W64_C' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Inception/GoogLeNet.yaml b/configs/Inception/GoogLeNet.yaml index 795bbcb13..87789678a 100644 --- a/configs/Inception/GoogLeNet.yaml +++ b/configs/Inception/GoogLeNet.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "GoogLeNet" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Inception/InceptionV4.yaml b/configs/Inception/InceptionV4.yaml index 65c73264f..42bcd0cd8 100644 --- a/configs/Inception/InceptionV4.yaml +++ b/configs/Inception/InceptionV4.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'InceptionV4' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV1/MobileNetV1.yaml b/configs/MobileNetV1/MobileNetV1.yaml index ff2b62b31..4f31699d2 100644 --- a/configs/MobileNetV1/MobileNetV1.yaml +++ b/configs/MobileNetV1/MobileNetV1.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV1" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV1/MobileNetV1_x0_25.yaml b/configs/MobileNetV1/MobileNetV1_x0_25.yaml index 12943b750..6cd83247d 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_25.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_25.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV1_x0_25" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV1/MobileNetV1_x0_5.yaml b/configs/MobileNetV1/MobileNetV1_x0_5.yaml index 14baaf658..a54b0743c 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_5.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV1_x0_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV1/MobileNetV1_x0_75.yaml b/configs/MobileNetV1/MobileNetV1_x0_75.yaml index 3563e1d95..d9841d850 100644 --- a/configs/MobileNetV1/MobileNetV1_x0_75.yaml +++ b/configs/MobileNetV1/MobileNetV1_x0_75.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV1_x0_75" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2.yaml b/configs/MobileNetV2/MobileNetV2.yaml index ba25a430d..31895f705 100644 --- a/configs/MobileNetV2/MobileNetV2.yaml +++ b/configs/MobileNetV2/MobileNetV2.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2_x0_25.yaml b/configs/MobileNetV2/MobileNetV2_x0_25.yaml index 25957a91b..a2b9ae894 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_25.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_25.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2_x0_25" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2_x0_5.yaml b/configs/MobileNetV2/MobileNetV2_x0_5.yaml index 4591353e6..9361c4ce3 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_5.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2_x0_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2_x0_75.yaml b/configs/MobileNetV2/MobileNetV2_x0_75.yaml index 757c87831..75ca6f32c 100644 --- a/configs/MobileNetV2/MobileNetV2_x0_75.yaml +++ b/configs/MobileNetV2/MobileNetV2_x0_75.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2_x0_75" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2_x1_5.yaml b/configs/MobileNetV2/MobileNetV2_x1_5.yaml index f23634721..3eeaae742 100644 --- a/configs/MobileNetV2/MobileNetV2_x1_5.yaml +++ b/configs/MobileNetV2/MobileNetV2_x1_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2_x1_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV2/MobileNetV2_x2_0.yaml b/configs/MobileNetV2/MobileNetV2_x2_0.yaml index 39996f76f..cc5358c1e 100644 --- a/configs/MobileNetV2/MobileNetV2_x2_0.yaml +++ b/configs/MobileNetV2/MobileNetV2_x2_0.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV2_x2_0" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml index bc27a07f3..32d168c80 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_large_x0_35" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml index 1aa847924..b3580040d 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_large_x0_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml index 3d859e2dd..e4bcfbac6 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_large_x0_75" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml index 32d0fe6fa..0477d1a3e 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_large_x1_0" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml index a368b8d02..00670ad1d 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_large_x1_25" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml index 7fee09f2c..afec8763e 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_small_x0_35" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml index 4659bd52f..9c7d3e334 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_small_x0_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml index 23d13b0ef..4febf84dc 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_small_x0_75" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml index f6369ec32..da4690484 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_small_x1_0" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml index cb711f845..c8fd74675 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "MobileNetV3_small_x1_25" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Res2Net/Res2Net101_vd_26w_4s.yaml b/configs/Res2Net/Res2Net101_vd_26w_4s.yaml index 2d5cecfd9..497f2bfa9 100644 --- a/configs/Res2Net/Res2Net101_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net101_vd_26w_4s.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'Res2Net101_vd_26w_4s' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Res2Net/Res2Net200_vd_26w_4s.yaml b/configs/Res2Net/Res2Net200_vd_26w_4s.yaml index 5cd51eb7b..cd57c30bb 100644 --- a/configs/Res2Net/Res2Net200_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net200_vd_26w_4s.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'Res2Net200_vd_26w_4s' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Res2Net/Res2Net50_14w_8s.yaml b/configs/Res2Net/Res2Net50_14w_8s.yaml index 69d249670..98e823424 100644 --- a/configs/Res2Net/Res2Net50_14w_8s.yaml +++ b/configs/Res2Net/Res2Net50_14w_8s.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'Res2Net50_14w_8s' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Res2Net/Res2Net50_26w_4s.yaml b/configs/Res2Net/Res2Net50_26w_4s.yaml index 2565bfcb1..8eace214b 100644 --- a/configs/Res2Net/Res2Net50_26w_4s.yaml +++ b/configs/Res2Net/Res2Net50_26w_4s.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'Res2Net50_26w_4s' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/Res2Net/Res2Net50_vd_26w_4s.yaml b/configs/Res2Net/Res2Net50_vd_26w_4s.yaml index 9aa79156c..aac1dcaec 100644 --- a/configs/Res2Net/Res2Net50_vd_26w_4s.yaml +++ b/configs/Res2Net/Res2Net50_vd_26w_4s.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'Res2Net50_vd_26w_4s' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt101_32x4d.yaml b/configs/ResNeXt/ResNeXt101_32x4d.yaml index 08c364894..4088cc8c2 100644 --- a/configs/ResNeXt/ResNeXt101_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt101_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt101_64x4d.yaml b/configs/ResNeXt/ResNeXt101_64x4d.yaml index 8a662284f..091d04aea 100644 --- a/configs/ResNeXt/ResNeXt101_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt101_64x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml index 4a70e2e02..40b331464 100644 --- a/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_vd_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt101_vd_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml index 1587b425d..b9cc0b95e 100644 --- a/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt101_vd_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt101_vd_64x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt152_32x4d.yaml b/configs/ResNeXt/ResNeXt152_32x4d.yaml index d073066eb..287013d27 100644 --- a/configs/ResNeXt/ResNeXt152_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt152_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt152_64x4d.yaml b/configs/ResNeXt/ResNeXt152_64x4d.yaml index 4cf492e25..27f0919cd 100644 --- a/configs/ResNeXt/ResNeXt152_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt152_64x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml index 5d89b7e24..341faca16 100644 --- a/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_vd_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt152_vd_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml index 877f4b6c9..ea37b955a 100644 --- a/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt152_vd_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt152_vd_64x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt50_32x4d.yaml b/configs/ResNeXt/ResNeXt50_32x4d.yaml index f8a7e8dd9..21efc0ad0 100644 --- a/configs/ResNeXt/ResNeXt50_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt50_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt50_64x4d.yaml b/configs/ResNeXt/ResNeXt50_64x4d.yaml index 4a5bf99da..2887a3425 100644 --- a/configs/ResNeXt/ResNeXt50_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ResNeXt50_64x4d" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml index b779b0524..1af4f5282 100644 --- a/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_vd_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ResNeXt50_vd_32x4d" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml b/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml index b79a63513..ea17291a8 100644 --- a/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml +++ b/configs/ResNeXt/ResNeXt50_vd_64x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNeXt50_vd_64x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet101.yaml b/configs/ResNet/ResNet101.yaml index 0ccbb13e4..5f2f398a0 100644 --- a/configs/ResNet/ResNet101.yaml +++ b/configs/ResNet/ResNet101.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet101' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet101_vd.yaml b/configs/ResNet/ResNet101_vd.yaml index c74b5b20b..c29df8da1 100644 --- a/configs/ResNet/ResNet101_vd.yaml +++ b/configs/ResNet/ResNet101_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet101_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet152.yaml b/configs/ResNet/ResNet152.yaml index c7934a040..d1918b300 100644 --- a/configs/ResNet/ResNet152.yaml +++ b/configs/ResNet/ResNet152.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet152' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet152_vd.yaml b/configs/ResNet/ResNet152_vd.yaml index fbf08ede3..b200663d2 100644 --- a/configs/ResNet/ResNet152_vd.yaml +++ b/configs/ResNet/ResNet152_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet152_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet18.yaml b/configs/ResNet/ResNet18.yaml index 270cd8ed3..033be21a1 100644 --- a/configs/ResNet/ResNet18.yaml +++ b/configs/ResNet/ResNet18.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet18' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet18_vd.yaml b/configs/ResNet/ResNet18_vd.yaml index 54f36b0f4..690ce3337 100644 --- a/configs/ResNet/ResNet18_vd.yaml +++ b/configs/ResNet/ResNet18_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet18_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet200_vd.yaml b/configs/ResNet/ResNet200_vd.yaml index f2004fdef..691ca702f 100644 --- a/configs/ResNet/ResNet200_vd.yaml +++ b/configs/ResNet/ResNet200_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet200_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet34.yaml b/configs/ResNet/ResNet34.yaml index cfe715d87..2ca0c702f 100644 --- a/configs/ResNet/ResNet34.yaml +++ b/configs/ResNet/ResNet34.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet34' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet34_vd.yaml b/configs/ResNet/ResNet34_vd.yaml index 39b9a3556..5b086d428 100644 --- a/configs/ResNet/ResNet34_vd.yaml +++ b/configs/ResNet/ResNet34_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet34_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet50.yaml b/configs/ResNet/ResNet50.yaml index 1fb825b58..8150e05f0 100644 --- a/configs/ResNet/ResNet50.yaml +++ b/configs/ResNet/ResNet50.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet50' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet50_vc.yaml b/configs/ResNet/ResNet50_vc.yaml index 233f00ce1..59cd1a220 100644 --- a/configs/ResNet/ResNet50_vc.yaml +++ b/configs/ResNet/ResNet50_vc.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet50_vc' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet/ResNet50_vd.yaml b/configs/ResNet/ResNet50_vd.yaml index dbb52e32b..b0b68b979 100644 --- a/configs/ResNet/ResNet50_vd.yaml +++ b/configs/ResNet/ResNet50_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'ResNet50_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ResNet_ACNet/ResNet_ACNet.yaml b/configs/ResNet_ACNet/ResNet_ACNet.yaml index 309f3821e..fa7fcf802 100644 --- a/configs/ResNet_ACNet/ResNet_ACNet.yaml +++ b/configs/ResNet_ACNet/ResNet_ACNet.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ResNet_ACNet" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SENet154_vd.yaml b/configs/SENet/SENet154_vd.yaml index 72adc722f..f60595719 100644 --- a/configs/SENet/SENet154_vd.yaml +++ b/configs/SENet/SENet154_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SENet154_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNeXt101_32x4d.yaml b/configs/SENet/SE_ResNeXt101_32x4d.yaml index bd9f20e7b..1a9887b53 100644 --- a/configs/SENet/SE_ResNeXt101_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt101_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNeXt101_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNeXt50_32x4d.yaml b/configs/SENet/SE_ResNeXt50_32x4d.yaml index c2a766013..1200d87b8 100644 --- a/configs/SENet/SE_ResNeXt50_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt50_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNeXt50_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml b/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml index f0adfb4b1..35f25500c 100644 --- a/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml +++ b/configs/SENet/SE_ResNeXt50_vd_32x4d.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNeXt50_vd_32x4d' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNet18_vd.yaml b/configs/SENet/SE_ResNet18_vd.yaml index 9684c6b51..007a179df 100644 --- a/configs/SENet/SE_ResNet18_vd.yaml +++ b/configs/SENet/SE_ResNet18_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNet18_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNet34_vd.yaml b/configs/SENet/SE_ResNet34_vd.yaml index 1ffe543dd..3ba4a804b 100644 --- a/configs/SENet/SE_ResNet34_vd.yaml +++ b/configs/SENet/SE_ResNet34_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNet34_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SENet/SE_ResNet50_vd.yaml b/configs/SENet/SE_ResNet50_vd.yaml index 8ca11f271..497a7f37b 100644 --- a/configs/SENet/SE_ResNet50_vd.yaml +++ b/configs/SENet/SE_ResNet50_vd.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: 'SE_ResNet50_vd' pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2.yaml b/configs/ShuffleNet/ShuffleNetV2.yaml index 5993e3f8c..5f4b4e83f 100644 --- a/configs/ShuffleNet/ShuffleNetV2.yaml +++ b/configs/ShuffleNet/ShuffleNetV2.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_swish.yaml b/configs/ShuffleNet/ShuffleNetV2_swish.yaml index e8ee5f446..e3e55a6ed 100644 --- a/configs/ShuffleNet/ShuffleNetV2_swish.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_swish.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_swish" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml index 9ab65f0d8..f80640b55 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_x0_25" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml index 134d5b9f3..fe13a98d6 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_x0_33" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml index 120ea1c11..85f930603 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_x0_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml index c1fc3d18c..3fb7dac9b 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_x1_5" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml index b45b70b64..a68d79517 100644 --- a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml +++ b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "ShuffleNetV2_x2_0" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SqueezeNet/SqueezeNet1_0.yaml b/configs/SqueezeNet/SqueezeNet1_0.yaml index 163bb33aa..4edb704e0 100644 --- a/configs/SqueezeNet/SqueezeNet1_0.yaml +++ b/configs/SqueezeNet/SqueezeNet1_0.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "SqueezeNet1_0" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/SqueezeNet/SqueezeNet1_1.yaml b/configs/SqueezeNet/SqueezeNet1_1.yaml index 4b716bc59..891d88a2b 100644 --- a/configs/SqueezeNet/SqueezeNet1_1.yaml +++ b/configs/SqueezeNet/SqueezeNet1_1.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "SqueezeNet1_1" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/VGG/VGG11.yaml b/configs/VGG/VGG11.yaml index d1cd5fab1..f1d8ab34c 100644 --- a/configs/VGG/VGG11.yaml +++ b/configs/VGG/VGG11.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "VGG11" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/VGG/VGG13.yaml b/configs/VGG/VGG13.yaml index 732695a15..cae3b7d58 100644 --- a/configs/VGG/VGG13.yaml +++ b/configs/VGG/VGG13.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "VGG13" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/VGG/VGG16.yaml b/configs/VGG/VGG16.yaml index 78f46b7f9..26f451e9b 100644 --- a/configs/VGG/VGG16.yaml +++ b/configs/VGG/VGG16.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "VGG16" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/VGG/VGG19.yaml b/configs/VGG/VGG19.yaml index 94aea056d..21da446ff 100644 --- a/configs/VGG/VGG19.yaml +++ b/configs/VGG/VGG19.yaml @@ -1,7 +1,7 @@ mode: 'train' architecture: "VGG19" pretrained_model: "" -model_save_dir: "./checkpoints/" +model_save_dir: "./output/" classes_num: 1000 total_images: 1281167 save_interval: 1 diff --git a/configs/eval.yaml b/configs/eval.yaml index 8edec558b..d9404f98b 100644 --- a/configs/eval.yaml +++ b/configs/eval.yaml @@ -6,7 +6,6 @@ total_images: 1281167 topk: 5 image_shape: [3, 224, 224] - VALID: batch_size: 16 num_workers: 4 From 0ce8c3912bcfb0a053adb6f0edfec1dc4ec9ce27 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 00:11:56 +0800 Subject: [PATCH 02/14] add gitignore --- .gitignore | 1 + 1 file changed, 1 insertion(+) diff --git a/.gitignore b/.gitignore index 3904a7ed8..cf9d572ca 100644 --- a/.gitignore +++ b/.gitignore @@ -3,6 +3,7 @@ *log* /dataset checkpoints/ +output/ pretrained/ *.ipynb* build/ From b86dfb4d79bfa2ea2cc46a0b9e6707ef9ca37a82 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 00:41:14 +0800 Subject: [PATCH 03/14] remove the bug run.sh --- tools/run.sh | 22 ---------------------- 1 file changed, 22 deletions(-) delete mode 100755 tools/run.sh diff --git a/tools/run.sh b/tools/run.sh deleted file mode 100755 index d718a3873..000000000 --- a/tools/run.sh +++ /dev/null @@ -1,22 +0,0 @@ -#!/usr/bin/env bash - -export PYTHONPATH=$(dirname "$PWD"):$PWD:$PYTHONPATH - -#python download.py -a ResNet181 -p ./pretrained/ -d 1 - -#python download.py -a ResNet18 -p ./pretrained/ -d 1 - -#python download.py -a ResNet34 -p ./pretrained/ -d 0 - -#python -m paddle.distributed.launch --selected_gpus="0,1,2,3" --log_dir=mylog tools/train.py - -#python -m paddle.distributed.launch --selected_gpus="0,1,2,3" --log_dir=mylog ./eval.py - -python -m paddle.distributed.launch \ - --selected_gpus="0,1,2,3" \ - --log_dir=mylog \ - tools/train.py \ - -c configs/ResNet/ResNet50_vd.yaml \ - -o use_mix=0 \ - -o TRAIN.batch_size=128 \ - -o TRAIN.transforms.3.NormalizeImage.mean.2=0.4 From e964c9edd51a2712c0898cc0cf7baa8ae8fe1b73 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 01:20:33 +0800 Subject: [PATCH 04/14] fix the bug of save model --- ppcls/utils/save_load.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index bdf173eae..047c934b4 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -16,9 +16,10 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import errno import os -import tempfile import shutil +import tempfile import paddle import paddle.fluid as fluid @@ -32,8 +33,16 @@ def _mkdir_if_not_exist(path): """ mkdir if not exists """ - if not os.path.exists(os.path.join(path)): - os.makedirs(os.path.join(path)) + if not os.path.exists(path): + try: + os.makedirs(path) + except OSError as e: + if e.errno == errno.EEXIST and os.path.isdir(path): + logger.warning( + 'be happy if some process has already created {}'.format( + path)) + else: + raise OSError('failed to mkdir {}'.format(path)) def _load_state(path): From 40c65e952ba26804ab22ee25dd06ab570d00c09e Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 01:21:31 +0800 Subject: [PATCH 05/14] fix the bug of save model --- ppcls/utils/save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index 047c934b4..e68a7af41 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -31,7 +31,7 @@ __all__ = ['init_model', 'save_model'] def _mkdir_if_not_exist(path): """ - mkdir if not exists + mkdir if not exists, ignore the exception when multiprocess mkdir together """ if not os.path.exists(path): try: From c3b973d977ea907168d75f096da0de58b69b677e Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 01:22:14 +0800 Subject: [PATCH 06/14] fix the bug of save model --- ppcls/utils/save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index e68a7af41..302171767 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -42,7 +42,7 @@ def _mkdir_if_not_exist(path): 'be happy if some process has already created {}'.format( path)) else: - raise OSError('failed to mkdir {}'.format(path)) + raise OSError('Failed to mkdir {}'.format(path)) def _load_state(path): From 543350fcff9b4e77c72d271e710843f698497a96 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:04:24 +0800 Subject: [PATCH 07/14] fix the getting_started --- docs/zh_cn/tutorials/getting_started.md | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/docs/zh_cn/tutorials/getting_started.md b/docs/zh_cn/tutorials/getting_started.md index 15ba2fdcb..4453cc903 100644 --- a/docs/zh_cn/tutorials/getting_started.md +++ b/docs/zh_cn/tutorials/getting_started.md @@ -22,8 +22,8 @@ PaddleClas 提供模型训练与评估脚本：tools/train.py和tools/eval.py python -m paddle.distributed.launch \ --selected_gpus="0,1,2,3" \ --log_dir=log_ResNet50 \ - train.py \ - -c ./configs/ResNet/ResNet50.yaml \ + tools/train.py \ + -c ./configs/ResNet/ResNet50.yaml ``` - 输出日志示例如下： @@ -38,9 +38,9 @@ epoch:0 train step:13 loss:7.9561 top1:0.0156 top5:0.1094 lr:0 python -m paddle.distributed.launch \ --selected_gpus="0,1,2,3" \ --log_dir=log_ResNet50_vd \ - train.py \ + tools/train.py \ -c ./configs/ResNet/ResNet50_vd.yaml \ - -o use_mix=1 \ + -o use_mix=1 ``` @@ -56,7 +56,7 @@ epoch:0 train step:522 loss:1.6330 lr:0.100000 elapse:0.210 ### 2.2 模型评估 ```bash -python eval.py \ +python tools/eval.py \ -c ./configs/eval.yaml \ -o architecture="ResNet50_vd" \ -o pretrained_model=path_to_pretrained_models @@ -76,7 +76,7 @@ python tools/export_model.py \ ``` 之后，通过预测引擎进行推理 ```bash -python tools/predict.py \ +python tools/infer/predict.py \ -m model文件路径 \ -p params文件路径 \ -i 图片路径 \ From 1a9b190ae35a638005a4cfba93cf2f04077ddd8a Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:19:08 +0800 Subject: [PATCH 08/14] add the logger --- .gitignore | 1 + 1 file changed, 1 insertion(+) diff --git a/.gitignore b/.gitignore index cf9d572ca..6062e2989 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ +!ppcls/ *.pyc *.sw* *log* From f29bb526fc52f25d9d26195d16257ccb3c9fb6b4 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:35:22 +0800 Subject: [PATCH 09/14] add the logger --- .gitignore | 5 ++- ppcls/utils/logger.py | 73 +++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 75 insertions(+), 3 deletions(-) create mode 100644 ppcls/utils/logger.py diff --git a/.gitignore b/.gitignore index 6062e2989..53c72a699 100644 --- a/.gitignore +++ b/.gitignore @@ -1,8 +1,7 @@ -!ppcls/ *.pyc *.sw* -*log* -/dataset +*/workerlog* +dataset/ checkpoints/ output/ pretrained/ diff --git a/ppcls/utils/logger.py b/ppcls/utils/logger.py new file mode 100644 index 000000000..bcc8ebe6f --- /dev/null +++ b/ppcls/utils/logger.py @@ -0,0 +1,73 @@ +#copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve. +# +#Licensed under the Apache License, Version 2.0 (the "License"); +#you may not use this file except in compliance with the License. +#You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +#Unless required by applicable law or agreed to in writing, software +#distributed under the License is distributed on an "AS IS" BASIS, +#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +#See the License for the specific language governing permissions and +#limitations under the License. + +import os +import logging +import random + +DEBUG = logging.DEBUG #10 +INFO = logging.INFO #20 +WARN = logging.WARN #30 +ERROR = logging.ERROR #40 + + +class Logger(object): + """ + Logger + """ + + def __init__(self, level=DEBUG): + self.init(level) + + def init(self, level=DEBUG): + """ + init + """ + self._logger = logging.getLogger() + self._logger.setLevel(level) + + def info(self, fmt, *args): + """info""" + self._logger.info(fmt, *args) + + def warning(self, fmt, *args): + """warning""" + self._logger.warning(fmt, *args) + + def error(self, fmt, *args): + """error""" + self._logger.error(fmt, *args) + + +_logger = Logger() + + +def init(level=DEBUG): + """init for external""" + _logger.init(level) + + +def info(fmt, *args): + """info""" + _logger.info(fmt, *args) + + +def warning(fmt, *args): + """warn""" + _logger.warning(fmt, *args) + + +def error(fmt, *args): + """error""" + _logger.error(fmt, *args) From 749c3d4168d96dbd30ffdfc7b7a682c0ab6d850c Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:48:00 +0800 Subject: [PATCH 10/14] add dataset --- dataset/README.md | 1 + dataset/download_imagenet2012.sh | 40 ++++++++++++++++++++++++++++++++ 2 files changed, 41 insertions(+) create mode 100644 dataset/README.md create mode 100644 dataset/download_imagenet2012.sh diff --git a/dataset/README.md b/dataset/README.md new file mode 100644 index 000000000..c1a50767c --- /dev/null +++ b/dataset/README.md @@ -0,0 +1 @@ +#mannual diff --git a/dataset/download_imagenet2012.sh b/dataset/download_imagenet2012.sh new file mode 100644 index 000000000..3e6e0ce6d --- /dev/null +++ b/dataset/download_imagenet2012.sh @@ -0,0 +1,40 @@ +set -e +if [ "x${IMAGENET_USERNAME}" == x -o "x${IMAGENET_ACCESS_KEY}" == x ];then + echo "Please create an account on image-net.org." + echo "It will provide you a pair of username and accesskey to download imagenet data." + read -p "Username: " IMAGENET_USERNAME + read -p "Accesskey: " IMAGENET_ACCESS_KEY +fi + +root_url=http://www.image-net.org/challenges/LSVRC/2012/nnoupb +valid_tar=ILSVRC2012_img_val.tar +train_tar=ILSVRC2012_img_train.tar +train_folder=train/ +valid_folder=val/ + +echo "Download imagenet training data..." +mkdir -p ${train_folder} +wget -nd -c ${root_url}/${train_tar} +tar xf ${train_tar} -C ${train_folder} + +cd ${train_folder} +for x in `ls *.tar` +do + filename=`basename $x .tar` + mkdir -p $filename + tar -xf $x -C $filename + rm -rf $x +done +cd - + +echo "Download imagenet validation data..." +mkdir -p ${valid_folder} +wget -nd -c ${root_url}/${valid_tar} +tar xf ${valid_tar} -C ${valid_folder} + +echo "Download imagenet label file: val_list.txt & train_list.txt" +label_file=ImageNet_label.tgz +label_url=http://paddle-imagenet-models.bj.bcebos.com/${label_file} +wget -nd -c ${label_url} +tar zxf ${label_file} + From 6e23365d8700cdb40190cebeeb261b58263333c4 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:53:55 +0800 Subject: [PATCH 11/14] polish run.sh --- tools/run.sh | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 tools/run.sh diff --git a/tools/run.sh b/tools/run.sh new file mode 100644 index 000000000..950e0cd3a --- /dev/null +++ b/tools/run.sh @@ -0,0 +1,7 @@ +export PYTHONPATH=$(dirname "$PWD"):$PWD:$PYTHONPATH + +python -m paddle.distributed.launch \ + --selected_gpus="0,1,2,3" \ + --log_dir=log_ResNet50 \ + tools/train.py \ + -c ./configs/ResNet/ResNet50.yaml From 06032aad1ce0b51a78ce80a34c09dc959bae42ee Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:54:37 +0800 Subject: [PATCH 12/14] polish run.sh --- tools/run.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/run.sh b/tools/run.sh index 950e0cd3a..14fd4ecd4 100644 --- a/tools/run.sh +++ b/tools/run.sh @@ -1,3 +1,5 @@ +#!/usr/bin/env bash + export PYTHONPATH=$(dirname "$PWD"):$PWD:$PYTHONPATH python -m paddle.distributed.launch \ From 89ee0d28b550f3d88655069bbf159b56c25e3996 Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 02:56:07 +0800 Subject: [PATCH 13/14] polish run.sh --- tools/run.sh | 0 1 file changed, 0 insertions(+), 0 deletions(-) mode change 100644 => 100755 tools/run.sh diff --git a/tools/run.sh b/tools/run.sh old mode 100644 new mode 100755 From 81c09cb24c48025e4b42f2975d5460292b47181f Mon Sep 17 00:00:00 2001 From: WuHaobo Date: Sat, 11 Apr 2020 03:13:45 +0800 Subject: [PATCH 14/14] polish run.sh --- tools/run.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/run.sh b/tools/run.sh index 14fd4ecd4..25d6ca023 100755 --- a/tools/run.sh +++ b/tools/run.sh @@ -1,6 +1,6 @@ #!/usr/bin/env bash -export PYTHONPATH=$(dirname "$PWD"):$PWD:$PYTHONPATH +export PYTHONPATH=$PWD:$PYTHONPATH python -m paddle.distributed.launch \ --selected_gpus="0,1,2,3" \