feat: 添加保存模型的功能和推理脚本

2026-02-03 18:03:47 +08:00
parent f5e2eca809
commit 3465782256
3 changed files with 112 additions and 1 deletions
@@ -6,7 +6,7 @@ defaults:
 train:
  batch_size: 4       # 减小 batch size 方便调试
  lr: 1e-4
-  max_steps: 100
+  max_steps: 10
  log_freq: 10
  device: "cpu"
  num_workers: 0      # 调试设为0，验证通过后改为 2 或 4