diff --git a/config/train.yaml b/config/train.yaml
index e88a1e9..8f81d2f 100644
--- a/config/train.yaml
+++ b/config/train.yaml
@@ -8,9 +8,9 @@ epochs: 100
 # 每一类的占比权重，如果要让每一类的占比权重相同，为1.0即可
 weight:
   - 1.0
-  - 2.9
+  - 4.0
   - 1.0
-  - 1.8
+  - 2.2
 
 # 数据集存放位置
 root: