From 52d5cc8115bf7724b92c8f486a0008d91ac6f1d0 Mon Sep 17 00:00:00 2001
From: shippingwang <shipeng1108@163.com>
Date: Mon, 6 Jul 2020 09:12:54 +0000
Subject: [PATCH] fix

---
 configs/high_performance/dali.yaml | 51 ++++++------------------------
 1 file changed, 9 insertions(+), 42 deletions(-)

diff --git a/configs/high_performance/dali.yaml b/configs/high_performance/dali.yaml
index 799748e5..36e7de0b 100644
--- a/configs/high_performance/dali.yaml
+++ b/configs/high_performance/dali.yaml
@@ -1,6 +1,6 @@
 mode: 'train'
 ARCHITECTURE:
-    name: 'ResNet50_vd'
+    name: "MobileNetV1"
 
 pretrained_model: ""
 model_save_dir: "./output/"
@@ -9,18 +9,18 @@ total_images: 1281167
 save_interval: 1
 validate: True
 valid_interval: 1
-epochs: 200
+epochs: 120
 topk: 5
 image_shape: [3, 224, 224]
 
-use_mix: True
-ls_epsilon: 0.1
-use_dali: True        
+use_dali: True
 
 LEARNING_RATE:
-    function: 'Cosine'          
-    params:                   
-        lr: 0.1               
+   function: 'Piecewise'
+   params:
+       lr: 0.1
+       decay_epochs: [30, 60, 90]
+       gamma: 0.1
 
 OPTIMIZER:
     function: 'Momentum'
@@ -28,7 +28,7 @@ OPTIMIZER:
         momentum: 0.9
     regularizer:
         function: 'L2'
-        factor: 0.000070
+        factor: 0.00003
 
 TRAIN:
     batch_size: 256
@@ -36,24 +36,6 @@ TRAIN:
     file_list: "./dataset/ILSVRC2012/train_list.txt"
     data_dir: "./dataset/ILSVRC2012/"
     shuffle_seed: 0
-    transforms:
-        - DecodeImage:
-            to_rgb: True
-            to_np: False
-            channel_first: False
-        - RandCropImage:
-            size: 224
-        - RandFlipImage:
-            flip_code: 1
-        - NormalizeImage:
-            scale: 1./255.
-            mean: [0.485, 0.456, 0.406]
-            std: [0.229, 0.224, 0.225]
-            order: ''
-        - ToCHWImage:
-    mix:                       
-        - MixupOperator:    
-            alpha: 0.2      
 
 VALID:
     batch_size: 64
@@ -61,18 +43,3 @@ VALID:
     file_list: "./dataset/ILSVRC2012/val_list.txt"
     data_dir: "./dataset/ILSVRC2012/"
     shuffle_seed: 0
-    transforms:
-        - DecodeImage:
-            to_rgb: True
-            to_np: False
-            channel_first: False
-        - ResizeImage:
-            resize_short: 256
-        - CropImage:
-            size: 224
-        - NormalizeImage:
-            scale: 1.0/255.0
-            mean: [0.485, 0.456, 0.406]
-            std: [0.229, 0.224, 0.225]
-            order: ''
-        - ToCHWImage:
-- 
GitLab