Merge pull request donnyyou#123 from donnyyou/master

fix cls bug.
nsl2014fm · Nov 27, 2019 · 3a19601 · 3a19601
2 parents 84f6212 + 1210109
commit 3a19601
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/configs/cls/cifar/base_cifar10_cls.conf b/configs/cls/cifar/base_cifar10_cls.conf
@@ -18,12 +18,12 @@
     "train": {
       "batch_size": 128,
       "aug_trans": {
-        "trans_seq": ["random_hflip", "padding", "random_crop"],
+        "trans_seq": ["random_hflip", "random_border", "random_crop"],
         "random_hflip": {
           "ratio": 0.5,
           "swap_pair": []
         },
-        "padding": {
+        "random_border": {
           "ratio": 1.0,
           "pad": [4, 4, 4, 4],
           "allow_outside_center": false
@@ -62,7 +62,8 @@
                      [255, 0, 255], [255, 0, 170], [255, 0, 85], [255, 0, 170], [255, 0, 85]]
     },
     "network":{
-      "model_name": "vgg19",
+      "model_name": "cls_model",
+      "backbone": "vgg19",
       "pooled_size": [1, 1],
       "checkpoints_name": "ic_vgg19_cifar10_cls",
       "checkpoints_dir": "checkpoints/cls/cifar10"
@@ -94,7 +95,7 @@
           "nesterov": false
         }
       },
-      "display_iters": 20,
+      "display_iter": 20,
       "save_iters": 2000,
       "test_interval": 100,
       "max_epoch": 360