Skip to content

Cosmos-Policy-ALOHA-Predict2-2B ​

Cosmos-Policy-ALOHA-Predict2-2B

nvidia 在 Hugging Face 发布的机器人模型,架构为 cosmos-policy,参数规模 2B,训练数据包含 nvidia/ALOHA-Cosmos-Policy。

模型英伟达nvidiacosmos-policy-aloha-predict2-2b

模型档案汇总架构、许可、适配机型与资料来源

Cosmos-Policy-ALOHA-Predict2-2B

nvidia 在 Hugging Face 发布的机器人模型,架构为 cosmos-policy,参数规模 2B,训练数据包含 nvidia/ALOHA-Cosmos-Policy。

  • 具身模型
世界模型研究2B英伟达更新于 8 个月前

概述

这个模型的核心架构、动作空间设计与真机部署定位。

Cosmos-Policy-ALOHA-Predict2-2B 是 NVIDIA 面向真实双臂机器人操作的 2B 参数策略模型,由 NVIDIA Cosmos-Predict2-2B-Video2World 视频基础模型单阶段微调而来,架构保持为带潜视频扩散的 diffusion transformer。模型输入自然语言任务描述、多视角 RGB 图像和本体状态,输出动作序列、未来状态预测与价值估计。

模型训练于 ALOHA-Cosmos-Policy 数据集,包含 ALOHA 2 平台、两台 ViperX 300 S 机械臂上的 185 条真实人类遥操作示范,覆盖 put X on plate、fold shirt、put candies in bowl、put candy in ziploc bag 四个双臂任务,并报告在这四项真实任务上的平均完成率为 93.6%。策略要求 25 Hz 控制器,使用顶视角和左右腕部相机,动作块为 50 步。

该模型用于研究与开发,许可证为 NVIDIA One-Way Noncommercial License,明确不可商用。推理集成提到 Transformers、Linux、NVIDIA Hopper,并仅测试过 BF16;部署到真实硬件时需注意相机、标定、关节限制和夹爪等硬件差异,并在首次执行时检查当前状态与下一步指令差异以保障安全。

资料出处 Hugging Face nvidia/Cosmos-Policy-ALOHA-Predict2-2B

关键参数与架构规格

参数量、网络主干、控制频率、显存需求及 Hugging Face Hub 统计。

参数量
2B
闭环控制频率
25Hz
Hub 下载量
209
出处Hugging Face
Hub 点赞
8
出处Hugging Face
网络架构
cosmos-policy
动作空间表示
连续关节位置/末端位姿动作空间
支持数据格式
SafeTensors

暂未公开或待收录:微调显存需求

快速部署与许可说明

开源许可证、权重拉取命令、预训练数据集与关联文章。

代码许可证

Apache-2.0

权重许可证

Apache-2.0

权重下载与策略推理参考

# 1. 从 Hugging Face Hub 下载模型权重与配置
huggingface-cli download nvidia/Cosmos-Policy-ALOHA-Predict2-2B --local-dir ./checkpoints/nvidia-cosmos-policy-aloha-predict2-2b

# 2. 使用 LeRobot / Transformers 加载策略开展真机或仿真闭环推理
python -m lerobot.scripts.eval --policy.path=nvidia/Cosmos-Policy-ALOHA-Predict2-2B

预训练与微调基准数据集:nvidia/ALOHA-Cosmos-Policy

关联机型

模型卡或论文声明在这些机器人上训练、微调或部署过;口径是作者声明,不是第三方复现。

没有已核实的适配记录。

同家族模型

同一家族里的热门模型。

来源

本页事实对应的公开页面,以原文为准。

  1. Hugging Face nvidia/Cosmos-Policy-ALOHA-Predict2-2B访问 2026-10-21