train: 更新类别权重设置

- 从配置中移除了特定于绝缘子的类别权重设置
- 替换为更通用的类别权重配置，增加了一个类别权重- 保持了权重设置的注释形式，以便未来可能的启用

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,              # 增加训练轮数