Skip to content

Commit cc09598

Browse files
authored
Adds instructions for using pre-made locomanipulation SDG dataset/model (#5357)
# Description Adds two tip callouts to the Demo 3 (G1 locomanipulation) section of the humanoids imitation learning documentation, giving users shortcuts to skip expensive pipeline steps: 1. Pre-made dataset (nvidia/g1_locomanip_dataset on Hugging Face): placed at the start of the SDG generation section, allowing users to skip manipulation dataset generation, SDG generation, and LeRobot conversion, and proceed directly to finetuning. Includes huggingface-cli download + unzip commands with the exact extracted path (g1_simple_high_var_lerobot/) and a note that policies trained on this dataset require --policy_quat_format wxyz at rollout. 2. Pre-trained model (nvidia/g1_locomanip_finetune on Hugging Face): placed immediately before the rollout section, allowing users to skip finetuning entirely. Includes download + unzip commands with the exact checkpoint path (g1_locomanip_finetune_20260129_231610/checkpoint-20000) and the required --policy_quat_format wxyz flag. Fixes # (issue) ## Type of change - Documentation update ## Screenshots ## Checklist - [x] I have read and understood the [contribution guidelines](https://isaac-sim.github.io/IsaacLab/main/source/refs/contributing.html) - [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with `./isaaclab.sh --format` - [x] I have made corresponding changes to the documentation - [x] My changes generate no new warnings - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] I have updated the changelog and the corresponding version in the extension's `config/extension.toml` file - [ ] I have added my name to the `CONTRIBUTORS.md` or my name already exists there
1 parent 16d6315 commit cc09598

1 file changed

Lines changed: 35 additions & 0 deletions

File tree

docs/source/overview/imitation-learning/humanoids_imitation.rst

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -538,6 +538,24 @@ Generate the dataset with manipulation and point-to-point navigation
538538

539539
To create a comprehensive locomanipulation dataset that combines both manipulation and navigation capabilities, you can generate a navigation dataset using the manipulation dataset from the previous step as input.
540540

541+
.. tip::
542+
543+
**Skip data generation:** A pre-made locomanipulation dataset in LeRobot format is available on
544+
Hugging Face at `nvidia/g1_locomanip_dataset <https://huggingface.co/datasets/nvidia/g1_locomanip_dataset>`__.
545+
Downloading it lets you skip this section and the dataset conversion step, proceeding directly to
546+
**Finetune the policy** below.
547+
548+
Download and unzip the dataset:
549+
550+
.. code:: bash
551+
552+
huggingface-cli download nvidia/g1_locomanip_dataset --repo-type dataset --local-dir ./datasets/g1_locomanip_hf
553+
unzip ./datasets/g1_locomanip_hf/*.zip -d ./datasets/
554+
555+
The archive extracts to ``./datasets/g1_simple_high_var_lerobot/``.
556+
Use this path as the ``--dataset-path`` in the finetuning step.
557+
Policies trained on this dataset require ``--policy_quat_format wxyz`` at rollout time.
558+
541559
.. list-table::
542560
:widths: 50 50
543561
:header-rows: 0
@@ -690,6 +708,23 @@ Run finetuning from the **Isaac-GR00T** repository root. Use the LeRobot-format
690708
691709
See the GR00T N1.5 repository documentation for additional training options.
692710

711+
.. tip::
712+
713+
**Skip finetuning:** A pre-trained GR00T N1.5 checkpoint for this task is available on
714+
Hugging Face at `nvidia/g1_locomanip_finetune <https://huggingface.co/nvidia/g1_locomanip_finetune>`__.
715+
Downloading it lets you skip the finetuning step and proceed directly to rollout.
716+
717+
Download and unzip the checkpoint:
718+
719+
.. code:: bash
720+
721+
huggingface-cli download nvidia/g1_locomanip_finetune --local-dir ./checkpoints/g1_locomanip_finetune_hf
722+
unzip ./checkpoints/g1_locomanip_finetune_hf/*.zip -d ./checkpoints/
723+
724+
The archive extracts to ``./checkpoints/g1_locomanip_finetune_20260129_231610/``.
725+
Use ``./checkpoints/g1_locomanip_finetune_20260129_231610/checkpoint-20000`` as the ``--model_path``
726+
in the rollout command below. This checkpoint requires ``--policy_quat_format wxyz``.
727+
693728
Rollout the policy in Isaac Lab
694729
"""""""""""""""""""""""""""""""
695730

0 commit comments

Comments
 (0)