更改默认参数

2026-02-11 17:14:32 +08:00
parent 320369ffb8
commit aba8779671
3 changed files with 6 additions and 6 deletions
@@ -9,19 +9,19 @@ defaults:
 # ====================
 train:
  # 基础训练参数
-  batch_size: 8       # 批次大小
+  batch_size: 32       # 批次大小
  lr: 1e-4            # 学习率
  max_steps: 100000    # 最大训练步数
  device: "cuda"      # 设备: "cuda" 或 "cpu"

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

  # 日志和检查点
  log_freq: 100       # 日志记录频率（步数）
-  save_freq: 5000     # 保存检查点频率（步数）
+  save_freq: 2000     # 保存检查点频率（步数）

  # 学习率调度器（带预热）
  warmup_steps: 500   # 预热步数