外观
Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B
nvidia 在 Hugging Face 发布的机器人模型,架构为 cosmos-policy-planning,参数规模 2B。
这个模型的核心架构、动作空间设计与真机部署定位。
Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B 是 NVIDIA 发布的 2B 参数规划检查点,用于真实 ALOHA 机器人环境中的双臂操作模型化规划。它不作为独立策略使用,而是与 Cosmos-Policy-ALOHA-Predict2-2B 配合:基础策略生成候选动作块,本模型对每个动作序列预测未来状态和值,并按预测价值选择动作。
模型采用与基础 ALOHA Predict2-2B 相同的潜在视频扩散 diffusion transformer 架构,并在 648 个策略 rollout episode 上微调,数据包含成功和失败轨迹,标签由策略执行和环境评估自动确定。输入包括任务文本、多视角 RGB 图像、14 维本体状态和 50 步 14 维动作序列;输出为 t+50 的三路图像、本体状态和值标量。
该检查点面向高精度、接触丰富且需要避免错误的 ALOHA 操作任务;用于规划时在挑战性 ALOHA 任务上带来 12.5 个百分点平均得分提升。模型要求特定相机配置和 25 Hz 控制频率,推理仅测试过 BF16,规划模式建议多 GPU 并行;许可为 NVIDIA One-Way Noncommercial License,非商业使用。
资料出处 Hugging Face nvidia/Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B
参数量、网络主干、控制频率、显存需求及 Hugging Face Hub 统计。
暂未公开或待收录:微调显存需求
开源许可证、权重拉取命令、预训练数据集与关联文章。
代码许可证
Apache-2.0
权重许可证
Apache-2.0
# 1. 从 Hugging Face Hub 下载模型权重与配置
huggingface-cli download nvidia/Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B --local-dir ./checkpoints/nvidia-cosmos-policy-aloha-planning-model-predict2-2b
# 2. 使用 LeRobot / Transformers 加载策略开展真机或仿真闭环推理
python -m lerobot.scripts.eval --policy.path=nvidia/Cosmos-Policy-ALOHA-Planning-Model-Predict2-2B模型卡或论文声明在这些机器人上训练、微调或部署过;口径是作者声明,不是第三方复现。
没有已核实的适配记录。
同一家族里的热门模型。
本页事实对应的公开页面,以原文为准。