更新训练配置中的类别权重，调整特定类别的权重值

2024-10-15 20:47:03 +08:00 · 2024-10-15 20:47:03 +08:00 · 257818f86d
commit 257818f86d
parent d5012516b6
1 changed files with 2 additions and 2 deletions
--- 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: