debug(train): 在siglip和DiffusionHead下跑通训练流程

2026-02-03 17:42:32 +08:00
parent 3b58760469
commit f5e2eca809
11 changed files with 414 additions and 70 deletions
--- a/roboimi/vla/models/heads/init.py
+++ b/roboimi/vla/models/heads/init.py
@@ -1,9 +1,9 @@
 # # Action Head models
-# from .diffusion import DiffusionActionHead
+from .diffusion import DiffusionHead
 # from .act import ACTHead

-# __all__ = ["DiffusionActionHead", "ACTHead"]
+__all__ = ["DiffusionHead"]

-from .debug import DebugHead
+# from .debug import DebugHead

-__all__ = ["DebugHead"]
+# __all__ = ["DebugHead"]