jiashunwang GitHub avatar

Hybrid-Motion-Imitation

jiashunwang

Unofficial implementation of GfR and HIL hybrid motion imitation frameworks for Unitree G1 humanoid robot, enabling box climbing and box moving tasks via a two-stage training that combines reference tracking and general reinforcement learning.

Stars

41

7-day growth

No data

Forks

0

Open issues

0

License

Apache-2.0

Last updated

2026-07-27

AI repository intelligence
FR-AI / ANALYSIS

Why it is worth attention

Introduces a novel hybrid learning approach where the actor policy is conditioned on task goals rather than reference motions, allowing the learned behavior to generalize to new task conditions from a small set of reference trajectories.

Who it is for

  • Robotics researchers studying humanoid locomotion and manipulation
  • Engineers working on sim-to-real transfer for legged robots
  • Developers interested in hybrid imitation + RL frameworks
  • Students learning advanced robot learning techniques

Use cases

  • Teaching a humanoid robot to climb onto boxes of varying sizes and positions
  • Training a robot to move objects like boxes to target locations
  • Providing a baseline for comparing goal-conditioned RL vs. reference-based methods
  • Web-based remote evaluation of learned policies without local simulation viewer

Strengths

  • Actor does not observe reference motion, enabling transfer to new task conditions
  • Two-stage training (track then mix) allows generalization from small reference sets
  • Headless training and evaluation with Viser web visualization for remote access
  • Built on Holosoma, leveraging existing robust simulation infrastructure

Considerations

  • Unofficial implementation not endorsed by original authors; may differ from published methods
  • Only validated on Unitree G1 29-DoF; porting to other robots may require significant adaptation
  • Training requires Isaac Sim and specific dependencies (wandb, CUDA), limiting accessibility
  • Only two specific tasks (box climbing, box moving) are provided; no broader task coverage demonstrated

README quick start

Hybrid Motion Imitation

  Box Moving
  


  Box Climbing
  

This is an unofficial implementation and extension of GfR and HIL, built on top of Holosoma. It is not the official code release for either paper.

This release provides two tasks for the 29-DoF Unitree G1:

exp:g1-29dof-wbt-hybrid-climb
exp:g1-29dof-wbt-hybrid-object

The same hybrid learning framework is used for both box climbing and box moving. Its key ideas are:

  1. The actor does not observe the target reference motion. We find that this design can achieve high-quality motion tracking across a wide range of tasks; see the HIL paper for more details. The actor only observes the robot state and sparse task conditions or goals, while the reference motion is used to define the goals and tracking rewards. Because the policy is conditioned on task goals rather than a reference motion, the learned behavior can easily transfer to new task conditions.

  2. Training combines two types of environments.

    • In reference-tracking environments, goals and task conditions are extracted from the reference motion and the policy is trained with tracking rewards.
    • In general RL environments, goals and task conditions are randomized and the policy is trained with sparse goal-reaching rewards.
  3. One policy is jointly optimized on two types of environments. The actor always uses the same observation interface across track and general RL envs, while the critic receives task-specific privileged information. This allows behavior learned from a small reference set to generalize to a wide range of task conditions and goals.

Setup

Run from the repository root:

bash scripts/setup_isaacsim.sh
source scripts/source_isaacsim_setup.sh
wandb login

Training

Training uses two stages:

  1. Stage 1 trains track environments only.
  2. Stage 2 resumes Stage 1 and trains a 50/50 mix of track and gen environments.

Climb Stage 1

CUDA_VISIBLE_DEVICES=0 \
python src/holosoma/holosoma/train_agent.py \
  exp:g1-29dof-wbt-hybrid-climb \
  logger:wandb \
  --training.num-envs=8192 \
  --algo.config.num-learning-iterations=15000 \
  --simul

Description

Unofficial implementation and extension of GfR (RSS 2026) and HIL (TOG 2026)

Related repositories

Similar projects matched by category, topics, and programming language.

lopopolo
Featured
lopopolo GitHub avatar

harness-engineering

Harness Engineering is a methodology for improving coding agent outputs by carefully crafting the environment around them—providing curated context, tools, and executable constraints that encode an organization’s nonfunctional requirements and cumulative lessons.

AI & Machine LearningAI Agents
2,390
slvDev
Featured
slvDev GitHub avatar

esp32-ai

A 28.9 million parameter language model runs on an $8 ESP32-S3 microcontroller entirely on-device, generating simple stories at about 9.5 tokens per second.

AI & Machine LearningLarge Language Models
1,960
littledivy
Featured
littledivy GitHub avatar

mimic

mimic captures traffic from any iOS or web app and automatically generates a Python client library that lets you call the app's API like a regular library.

AI & Machine Learning
1,482