Skip to content

Commit 4698d2d

Browse files
authored
Improves dataset file names and low success rate for trained model on g1 locomanipulation dataset (#3503)
# Description Changes mimic data generation parameters for `LocomanipulationG1MimicEnvCfg` to increase action noise to match that of `PickPlaceGR1T2MimicEnvCfg`. Also, updates documentation for teleop_imitation.rst for dataset download links to point to the renamed files that matches the tutorial instructions. Fixes # (issue) <!-- As a practice, it is recommended to open an issue to have discussions on the proposed pull request. This makes it easier for the community to keep track of what is being developed or added, and if a given feature is demanded by more than one party. --> ## Type of change <!-- As you go through the list, delete the ones that are not applicable. --> - Bug fix (non-breaking change which fixes an issue) - Documentation update ## 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 - [x] I have added tests that prove my fix is effective or that my feature works - [x] I have updated the changelog and the corresponding version in the extension's `config/extension.toml` file - [x] I have added my name to the `CONTRIBUTORS.md` or my name already exists there <!-- As you go through the checklist above, you can mark something as done by putting an x character in it For example, - [x] I have done this task - [ ] I have not done this task -->
1 parent 3b794f9 commit 4698d2d

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ Generate the dataset
451451
^^^^^^^^^^^^^^^^^^^^
452452

453453
If you skipped the prior collection and annotation step, download the pre-recorded annotated dataset ``dataset_annotated_gr1.hdf5`` from
454-
here: `[Annotated GR1 Dataset] <https://omniverse-content-production.s3-us-west-2.amazonaws.com/Assets/Isaac/5.0/Isaac/IsaacLab/Mimic/pick_place_datasets/dataset_annotated_gr1_v01.hdf5>`_.
454+
here: `[Annotated GR1 Dataset] <https://omniverse-content-production.s3-us-west-2.amazonaws.com/Assets/Isaac/5.0/Isaac/IsaacLab/Mimic/pick_place_datasets/dataset_annotated_gr1.hdf5>`_.
455455
Place the file under ``IsaacLab/datasets`` and run the following command to generate a new dataset with 1000 demonstrations.
456456

457457
.. code:: bash
@@ -573,7 +573,7 @@ Follow the same data collection, annotation, and generation process as demonstra
573573
574574
575575
If you skipped the prior collection and annotation step, download the pre-recorded annotated dataset ``dataset_annotated_g1_locomanip.hdf5`` from
576-
here: `[Annotated G1 Dataset] <https://omniverse-content-production.s3-us-west-2.amazonaws.com/Assets/Isaac/5.0/Isaac/IsaacLab/Mimic/pick_place_datasets/dataset_annotated_g1_locomanip_v01.hdf5>`_.
576+
here: `[Annotated G1 Dataset] <https://omniverse-content-production.s3-us-west-2.amazonaws.com/Assets/Isaac/5.0/Isaac/IsaacLab/Mimic/pick_place_datasets/dataset_annotated_g1_locomanip.hdf5>`_.
577577
Place the file under ``IsaacLab/datasets`` and run the following command to generate a new dataset with 1000 demonstrations.
578578

579579
.. code:: bash

source/isaaclab_mimic/isaaclab_mimic/envs/pinocchio_envs/locomanipulation_g1_mimic_env_cfg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def __post_init__(self):
5252
# Optional parameters for the selection strategy function
5353
selection_strategy_kwargs={"nn_k": 3},
5454
# Amount of action noise to apply during this subtask
55-
action_noise=0.002,
55+
action_noise=0.003,
5656
# Number of interpolation steps to bridge to this subtask segment
5757
num_interpolation_steps=0,
5858
# Additional fixed steps for the robot to reach the necessary pose
@@ -74,7 +74,7 @@ def __post_init__(self):
7474
# Optional parameters for the selection strategy function
7575
selection_strategy_kwargs={"nn_k": 3},
7676
# Amount of action noise to apply during this subtask
77-
action_noise=0.002,
77+
action_noise=0.003,
7878
# Number of interpolation steps to bridge to this subtask segment
7979
num_interpolation_steps=3,
8080
# Additional fixed steps for the robot to reach the necessary pose
@@ -100,7 +100,7 @@ def __post_init__(self):
100100
# Optional parameters for the selection strategy function
101101
selection_strategy_kwargs={"nn_k": 3},
102102
# Amount of action noise to apply during this subtask
103-
action_noise=0.002,
103+
action_noise=0.003,
104104
# Number of interpolation steps to bridge to this subtask segment
105105
num_interpolation_steps=0,
106106
# Additional fixed steps for the robot to reach the necessary pose

0 commit comments

Comments
 (0)