跑通配置和训练脚本

2026-02-03 16:51:04 +08:00
parent bd8bbb0cfc
commit 3b58760469
5 changed files with 227 additions and 90 deletions
@@ -1,9 +1,12 @@
 defaults:
  - _self_
-  - agent: debug_vla  # <--- This tells Hydra to look in conf/agent/ and load debug_vla.yaml
-  # Future expansions:
-  # - data: robomimic_hdf5
-  # - train: standard
+  - agent: tiny
+  - data: custom_hdf5  # 新增这一行，激活数据配置

-# Global settings (optional for now)
-seed: 42
+train:
+  batch_size: 4       # 减小 batch size 方便调试
+  lr: 1e-4
+  max_steps: 100
+  log_freq: 10
+  device: "cpu"
+  num_workers: 0      # 调试设为0，验证通过后改为 2 或 4