stable-worldmodel
by galilai-group
介绍
stable-worldmodel 是由 galilai-group 开发的开源世界模型研究与评估平台。它为世界模型研究的核心三阶段(数据收集、模型训练、基于模型预测控制的评估)提供了统一的接口。该工具内置了多种规划求解器与参考基准(如LeWM和DINO-WM),支持LanceDB、HDF5及LeRobot等高效数据集格式。此外,它集成了涵盖机器人、雅达利等大量标准化环境,并提供视觉与物理属性的独立变异因子,极大简化了零样本泛化测试流程。
功能特征
- 统一的研究工作流(数据收集、训练与MPC评估)
- 高性能数据格式支持(内置LanceDB及LeRobot适配)
- 丰富且标准化的评估环境(支持多变异因子泛化测试)
- 开箱即用的参考基准(内置LeWM、DINO-WM及求解器)
支持平台
desktopiot