From 806675d2260fc5275ef3bea22c3c50aa95bd7efd Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Tue, 27 Oct 2020 17:36:44 +0800
Subject: [PATCH] fix config files (#344)

---
 configs/DarkNet/DarkNet53.yaml | 6 ++++++
 configs/VGG/VGG11.yaml         | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/configs/DarkNet/DarkNet53.yaml b/configs/DarkNet/DarkNet53.yaml
index 525ba711..1dfe7301 100644
--- a/configs/DarkNet/DarkNet53.yaml
+++ b/configs/DarkNet/DarkNet53.yaml
@@ -13,6 +13,9 @@ epochs: 200
 topk: 5
 image_shape: [3, 256, 256]
 
+use_mix: True
+ls_epsilon: 0.1
+
 LEARNING_RATE:
    function: 'Cosine'
    params:
@@ -47,6 +50,9 @@ TRAIN:
             std: [0.229, 0.224, 0.225]
             order: ''
         - ToCHWImage:
+    mix:
+        - MixupOperator:
+            alpha: 0.2
 
 VALID:
     batch_size: 64
diff --git a/configs/VGG/VGG11.yaml b/configs/VGG/VGG11.yaml
index 9331f52f..4f26a162 100644
--- a/configs/VGG/VGG11.yaml
+++ b/configs/VGG/VGG11.yaml
@@ -24,7 +24,7 @@ OPTIMIZER:
         momentum: 0.9
     regularizer:
         function: 'L2'
-        factor: 0.0004
+        factor: 0.0002
 
 TRAIN:
     batch_size: 512
-- 
GitLab