s

stable-worldmodel

by galilai-group
🔓 开源项目 Python 🌍 全球 free

介绍

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

功能特征

  • 统一的研究工作流(数据收集、训练与MPC评估)
  • 高性能数据格式支持(内置LanceDB及LeRobot适配)
  • 丰富且标准化的评估环境(支持多变异因子泛化测试)
  • 开箱即用的参考基准(内置LeWM、DINO-WM及求解器)

支持平台

desktopiot