feat: add IMF AttnRes policy training path

2026-04-01 23:35:31 +08:00
parent 8d6060224a
commit c2000b5533
10 changed files with 1566 additions and 11 deletions
@@ -13,6 +13,7 @@ train:
  lr: 1e-4            # 学习率
  max_steps: 100000    # 最大训练步数
  device: "cuda"      # 设备: "cuda" 或 "cpu"
+  disable_cudnn: false  # 遇到当前机器的 cuDNN 兼容性问题时可置 true

  # 数据加载
  num_workers: 12     # DataLoader 工作进程数（调试时设为 0）