添加pad_loss

2026-02-11 20:33:26 +08:00
parent eeb07cad15
commit 83cd55e67b
5 changed files with 27 additions and 8 deletions
@@ -9,13 +9,13 @@ defaults:
 # ====================
 train:
  # 基础训练参数
-  batch_size: 32       # 批次大小
+  batch_size: 8       # 批次大小
  lr: 1e-4            # 学习率
  max_steps: 100000    # 最大训练步数
  device: "cuda"      # 设备: "cuda" 或 "cpu"

  # 数据加载
-  num_workers: 40      # DataLoader 工作进程数（调试时设为 0，生产环境用 8）
+  num_workers: 8      # DataLoader 工作进程数（调试时设为 0，生产环境用 8）
  val_split: 0.1      # 验证集比例
  seed: 42            # 随机种子（用于数据划分）