Commit 667d221b authored by shauryadNv's avatar shauryadNv Committed by Kelly Guo

Fixes train and eval commands in Mimic Cosmos pipeline doc. (#512)

# Description

Fixes the train and evaluation commands specified in the Mimic Cosmos
pipeline doc to use the correct custom environment.

## Type of change

- Bug fix (non-breaking change which fixes an issue)
- This change requires a documentation update

## Checklist

- [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
- [x] I have added my name to the `CONTRIBUTORS.md` or my name already
exists there
parent eef56910
...@@ -293,12 +293,12 @@ To install the robomimic framework, use the following commands: ...@@ -293,12 +293,12 @@ To install the robomimic framework, use the following commands:
Training an agent Training an agent
^^^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^^^
Using the generated data, we can now train a visuomotor BC agent for ``Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-v0``: Using the generated data, we can now train a visuomotor BC agent for ``Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-Cosmos-v0``:
.. code:: bash .. code:: bash
./isaaclab.sh -p scripts/imitation_learning/robomimic/train.py \ ./isaaclab.sh -p scripts/imitation_learning/robomimic/train.py \
--task Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-v0 --algo bc \ --task Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-Cosmos-v0 --algo bc \
--dataset ./datasets/mimic_cosmos_dataset.hdf5 \ --dataset ./datasets/mimic_cosmos_dataset.hdf5 \
--name bc_rnn_image_franka_stack_mimic_cosmos --name bc_rnn_image_franka_stack_mimic_cosmos
...@@ -379,8 +379,8 @@ Example usage for the cube stacking task: ...@@ -379,8 +379,8 @@ Example usage for the cube stacking task:
.. code:: bash .. code:: bash
./isaaclab.sh -p scripts/imitation_learning/robomimic/robust_eval.py \ ./isaaclab.sh -p scripts/imitation_learning/robomimic/robust_eval.py \
--task Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-v0 \ --task Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-Cosmos-v0 \
--input_dir logs/robomimic/Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-v0/bc_rnn_image_franka_stack_mimic_cosmos/*/models \ --input_dir logs/robomimic/Isaac-Stack-Cube-Franka-IK-Rel-Visuomotor-Cosmos-v0/bc_rnn_image_franka_stack_mimic_cosmos/*/models \
--log_dir robust_results/bc_rnn_image_franka_stack_mimic_cosmos \ --log_dir robust_results/bc_rnn_image_franka_stack_mimic_cosmos \
--log_file result \ --log_file result \
--enable_cameras \ --enable_cameras \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment