Conversation
71f1e9a to
dbe0bac
Compare
|
I included all OSMO related cmds here. We can remove them later if you think it's too much. But at least for TME, they can follow the exact steps 1-5 to make sure at least closedloop works for each individual task. Then We can do the cross task training. Otherwise we will be suffering from not working trajectories. @alexmillane WDYT |
|
@peterd-NV in closedloop, the close door move rate metric is 0, which does not quite make sense to me. Can you take a look? |
alexmillane
left a comment
There was a problem hiding this comment.
Looks great! Thanks for doing this!
| .. todo:: | ||
|
|
||
| Verifying Arena-GR00T on Blackwell architecture. | ||
|
|
| * - **Interop** | ||
| - Isaac Lab Teleop, Isaac Lab Mimic | ||
| * - **Scene** | ||
| - Kitchen environment |
There was a problem hiding this comment.
Maybe we better put Light Wheel Robocasa Kitchen to attribute LightWheel.
| * - **Scene** | ||
| - Kitchen environment | ||
| * - **Objects** | ||
| - Refrigerator (articulated object) |
There was a problem hiding this comment.
- object to place in refrigerator (rigid object)
| Follow the instructions described on the CLI to mark subtask boundaries: | ||
|
|
||
| Left Arm: | ||
| 1. **Pick & place completed:** Robot has completed the pick and place task |
There was a problem hiding this comment.
Perhaps it's made clear in the CLI, but I'm wondering whether we should specify where the user should press the key here. Does this mean that you press the button once the door is closed? How does that interact with the reset.
There was a problem hiding this comment.
Adding this to docs
| Step 3: Validate Generated Data (Optional) | ||
| ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ | ||
|
|
||
| In order to validation the generated dataset, you can replay the generated data |
There was a problem hiding this comment.
nit: in order to validate
| To verify the generated LeRobot dataset, you first need to compute statistics information about the dataset. | ||
| This can be done by running the following command: | ||
|
|
||
| .. code-block:: bash | ||
|
|
||
| python submodules/Isaac-GR00T/gr00t/data/stats.py $DATASET_DIR/ranch_bottle_into_fridge/ranch_bottle_into_fridge_generated_100/lerobot gr1_joint |
There was a problem hiding this comment.
I notice that the stats command alters some files. Does this have an affect on training? I wasn't running this when training things in Zurich. I'm wondering if that could have affected the performance?
There was a problem hiding this comment.
stats.py added stats.json & rel_stats.json which I think it's used in dataloader for normalization. It's included in training but not replaying. I don't think it's a bug from stats ops itself, as it mostly copies from n1.5.
|
|
||
| .. todo:: | ||
| There is a bug in ISAAC-GR00T submodule gr00t/data/stats.py. PR is submmited and in review from Gear. | ||
| Usele gpu finetune command in the last step for now. |
| --dataset_path=$DATASET_DIR/arena_gr1_manipulation_dataset_generated/lerobot \ | ||
| --output_dir=$MODELS_DIR \ | ||
| --modality_config_path=isaaclab_arena_gr00t/embodiments/gr1/gr1_arms_only_data_config.py \ | ||
| --global_batch_size=24 \ |
There was a problem hiding this comment.
shall we go with 96? In my testing this results in a lower training error, and doesn't affect the training time.
There was a problem hiding this comment.
good catch. It was trained with 96 but I forgot to update the doc here
| python isaaclab_arena/evaluation/policy_runner.py \ | ||
| --policy_type isaaclab_arena_gr00t.policy.gr00t_closedloop_policy.Gr00tClosedloopPolicy \ | ||
| --policy_config_yaml_path isaaclab_arena_gr00t/policy/config/gr1_manip_ranch_bottle_gr00t_closedloop_config.yaml \ | ||
| --num_steps 2000 \ |
There was a problem hiding this comment.
To me it makes more sense that we run for a number of episodes. I'm adding that here: #383
What do you think?
There was a problem hiding this comment.
I think it's gonna be a great idea once you merge it. It looks like it has issue for recording metrics yesterday I ran it.
8d0cae1 to
103fb87
Compare
…tation for closing door
…tation for closing door
…rena into peterd/gtc_dli_docs
Summary
Add docs for GTC DLI course GR1 sequential static manipulation workflow