目录

LeRobot Hugging Face Hub 集成

本页面记录 LeRobot 如何与 Hugging Face Hub 集成以存储、版本化和分发数据集、策略和奖励模型。Hub 作为所有 LeRobot 组件的中央仓库,实现跨社区的无缝共享和可复现性。

LeRobot 利用 huggingface_hub 库为机器人数据提供 Git 风格的体验,支持版本化快照、选择性下载和自动元数据生成。

返回概述

Hub 集成架构

LeRobot 将 Hugging Face Hub 视为一等存储后端。集成主要由 LeRobotDataset(数据)和 HubMixin(通过 RewardModelConfigPreTrainedPolicy 的模型)处理。

数据集 Hub 操作

数据集下载使用 snapshot_download 并带有智能缓存:

'LeRobotDatasetMetadata' 类处理初始发现,先下载 meta/ 目录,允许系统在提交下载大型 Parquet 或视频文件前检查数据集结构。

数据集元数据文件

Hub 上的每个数据集仓库包含由 LeRobotDatasetMetadata 管理的 meta/ 目录:

模型和奖励 Hub 操作

LeRobot 使用 HubMixin 为配置类提供 Hub 功能:

版本控制和兼容性

认证和缓存

Hub 缓存

认证机制

相关页面