openvla-traj-head-dagger-r2

StevenZhou90/openvla-traj-head-dagger-r2

openvla-traj-head-dagger-r2 is a robotics model from StevenZhou90. Use it for the robotics task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.

This repository contains the DAgger round-2 trajectory draft head used for the OpenVLA + SimplerEnv fast trajectory policy.

Downloads · 30 days

Access

Public

Updated May 5, 2026

Repo size

41.1 MB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.pt41.1 MB · 100%

At a glance

Task: Robotics
License: apache-2.0
Access: Public
Created: May 5, 2026
Updated: May 5, 2026
SHA: 276b68cd

Model card

From the Hugging Face model README

OpenVLA Trajectory Head DAgger R2

This repository contains the DAgger round-2 trajectory draft head used for the OpenVLA + SimplerEnv fast trajectory policy.

Files

best.pt: TinyTrajectoryHead checkpoint
metrics.json: training metrics for the selected checkpoint

Benchmark Snapshot

Matched SimplerEnv benchmark matrix:

tasks: vertical/horizontal/standing coke-can
x positions: -0.3500, -0.2925, -0.2350
obj_init_y = -0.02
3 episodes per setting, 27 episodes total
hardware: 1x A100-SXM4-40GB, bfloat16

Decoder	Success	Avg ms/step	Speedup
Baseline OpenVLA	14/27 (51.9%)	295.2	1.00x
Adaptive fast policy (gate v=5, h=6, s=6)	16/27 (59.3%)	105.0	2.81x

Usage

Download best.pt to:

checkpoints/traj_head_dagger_r2/best.pt

Then run:

uv run python scripts/run_openvla_sim.py \
  --decoder trajectory-spec \
  --trajectory-head-checkpoint checkpoints/traj_head_dagger_r2/best.pt \
  --trajectory-fast-draft-only \
  --trajectory-head-threshold 0.2 \
  --trajectory-fast-min-confident-tokens 5 \
  --task google_robot_pick_vertical_coke_can \
  --published-eval-setup \
  --episodes 3 \
  --steps 80

For full reproduction from data generation through benchmarking, use:

./scripts/reproduce_readme_speedup_local.sh

Task: Robotics
License: apache-2.0
Created: May 5, 2026
Updated: May 5, 2026