Residual Copilot β€” Models

Policy checkpoints for Residual Shared Autonomy. Includes BC pilot models (DiffusionPolicy), RL residual copilot policies, and standalone autonomous baselines.

Contents

shared_autonomy_policies/
β”œβ”€β”€ bc_teleop/                            # DiffusionPolicy pilots trained on teleop data
β”‚   β”œβ”€β”€ GearMesh_bc_teleop/
β”‚   β”œβ”€β”€ PegInsert_bc_teleop/
β”‚   └── NutThread_bc_teleop/
β”œβ”€β”€ bc_expert/                            # DiffusionPolicy pilots trained on expert data
β”‚   β”œβ”€β”€ GearMesh_bc_expert/
β”‚   β”œβ”€β”€ PegInsert_bc_expert/
β”‚   └── NutThread_bc_expert/
└── residual_copilot/                     # RL residual copilot policies (RL-Games PPO)
    β”œβ”€β”€ {Task}_noisy_knn/nn/FactoryXarm.pth    # Trained with noisy kNN pilot
    └── {Task}_bc_teleop/nn/FactoryXarm.pth    # Trained with BC teleop pilot
autonomous_policies/                      # Standalone DiffusionPolicy baselines (no residual)
β”œβ”€β”€ dp_teleop_data_match_succ/
β”‚   └── gear_mesh_ckpt_30k/
β”œβ”€β”€ dp_teleop_data_match_trials/
β”‚   └── gear_mesh_ckpt_30k/
β”œβ”€β”€ dp_residual_data_match_succ/
β”‚   └── gear_mesh_ckpt_30k/
└── dp_residual_data_match_trials/
    └── gear_mesh_ckpt_30k/

Usage

Models are auto-downloaded at runtime via resolve_hf(). See the main repo for evaluation instructions.

# Evaluate residual copilot with noisy kNN pilot
python scripts/play.py --task GearMesh --pilot kNNPilot --copilot ResidualCopilot

Related

Repo Type Contents
residual_copilot_assets dataset Robot URDF/USD and object meshes
residual_copilot_data dataset Teleoperation training data (.npy)
Downloads last month

-

Downloads are not tracked for this model. How to track
Video Preview
loading

Collection including shashuo0104/residual_copilot_models