/ Модель

Характеристики

Библиотека
rlinf
Опубликована
1 авг 2026

pi05-stack-cubes-umi-step20000

Языковая модельИзображение, текст → action

Контрольная точка Pi0.5, дообученная для управления Franka при складывании кубиков и подготовленная для развёртывания и оценки в RLinf.

7скачиваний за 30 дней0лайковвеса обновлены1 авг 2026

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Это контрольная точка модели Pi0.5, дообученная методом SFT для задачи складывания кубиков с манипулятором Franka и экспортированная для развёртывания и оценки в RLinf. Чекпойнт соответствует шагу обучения 20 000 и предназначен для управления роботом в сценарии UMI single-Franka.

Модель принимает наблюдения среды и формирует действия: используется пространство действий TCP с представлением вращения 6D, размерность действия — 10, а RLinf применяет чанки из 20 действий. Веса сохранены в формате safetensors с точностью BF16; экспорт преобразован из формата обучения RLinf FSDP в формат PyTorch OpenPI. В репозитории также есть конфигурация архитектуры и статистики нормализации задачи.

Экспорт рассчитан на развёртывание и оценку, но не содержит состояния оптимизатора и загрузчика данных, поэтому продолжить обучение FSDP с него нельзя. Лицензия в предоставленных данных не указана.

Сценарии применения

  • Управление манипулятором Franka при складывании кубиков
  • Оценка политики Pi0.5 на задаче stack_cubes_umi_1_franka
  • Развёртывание роботизированной политики через RLinf
  • Генерация чанков действий в TCP-пространстве с вращением 6D

Ограничения

Экспорт предназначен для развёртывания и оценки и не содержит состояния оптимизатора или загрузчика данных, необходимых для возобновления обучения FSDP. Указанный сценарий ограничен задачей складывания кубиков с одним манипулятором Franka.

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

Pi0.5 UMI Single-Franka — SFT Step 20000

This is an RLinf deployment export of the global_step_20000 SFT checkpoint for the UMI single-Franka stack_cubes_umi_1_franka task.

The checkpoint was converted from the RLinf FSDP training format to the bare OpenPI PyTorch format expected by the RLinf real-world evaluator. All floating point tensors are stored as BF16.

Files

  • model.safetensors: Pi0.5 model weights.
  • config.json: Pi0.5 model architecture configuration.
  • norm_stats/norm_stats.json: task normalization statistics.

Download

hf download <OWNER>/<REPOSITORY> --local-dir /path/to/pi05_step_20000

For a private repository, authenticate first with hf auth login using an account that has read access.

RLinf deployment

From the RLinf repository root:

bash evaluations/run_umi_left_franka_eval.sh \
  --model_path /path/to/pi05_step_20000 \
  --norm_stats_path /path/to/pi05_step_20000/norm_stats \
  --wrist_camera <WRIST_CAMERA_SERIAL> \
  --front_camera <FRONT_CAMERA_SERIAL>

The evaluator also requires the Franka and OpenPI runtime environments described in the RLinf real-world evaluation documentation.

Checkpoint metadata

  • Training step: 20000
  • Policy: Pi0.5, single Franka, TCP rotation-6D action space
  • Action dimension: 10
  • Action chunk used by RLinf: 20
  • Export dtype: bfloat16
  • Number of tensors: 667

This export is intended for deployment and evaluation. It does not contain the optimizer or dataloader state required to resume FSDP training.

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · rlinf

Нельзя запустить

Модели для роботов не запускаются как сервис инференса.

Ревизия весов
b9310d8

0 моделей