From 21d190a658631e9e17a951b923fea5b77bbf38aa Mon Sep 17 00:00:00 2001
From: fly6516 <fly6516@outlook.com>
Date: Thu, 29 May 2025 20:23:21 +0800
Subject: [PATCH] =?UTF-8?q?train:=20=E6=9B=B4=E6=96=B0=E7=B1=BB=E5=88=AB?=
 =?UTF-8?q?=E6=9D=83=E9=87=8D=E8=AE=BE=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 从配置中移除了特定于绝缘子的类别权重设置
- 替换为更通用的类别权重配置，增加了一个类别权重- 保持了权重设置的注释形式，以便未来可能的启用
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 83e930c..57d6011 100644
--- a/train.py
+++ b/train.py
@@ -31,7 +31,7 @@ TRAIN_CONFIG = {
         "weight_decay": 0.001,     # 权重衰减
         "label_smoothing": 0.1,    # 标签平滑
         "mosaic": True,            # 启用Mosaic增强
-        # "class_weights": [1.0, 2.0]  # 设置类别权重，insulator权重更高
+        # "class_weights": [2.0, 2.0, 1.0]  # 设置类别权重
     },
     "stage2": {
         "epochs": 150,              # 增加训练轮数