feat(train): 跑通训练脚本

2026-02-05 14:08:43 +08:00
parent dd2749cb12
commit b0a944f7aa
17 changed files with 1002 additions and 464 deletions
--- a/roboimi/vla/models/heads/init.py
+++ b/roboimi/vla/models/heads/init.py
@@ -1,9 +1,8 @@
 # # Action Head models
-from .diffusion import DiffusionHead
+from .diffusion import ConditionalUnet1D
 # from .act import ACTHead

-__all__ = ["DiffusionHead"]
+__all__ = ["ConditionalUnet1D"]

 # from .debug import DebugHead
-
 # __all__ = ["DebugHead"]