Skip to content

README_train.md

Latest commit

 

History

History
52 lines (33 loc) · 1.99 KB

README_train.md

File metadata and controls

52 lines (33 loc) · 1.99 KB
Raw

Installation Train

  1. Install Isaac Lab by following the installation guide. We recommend using the conda installation as it simplifies calling Python scripts from the terminal.

  2. Install git for very large file

sudo apt install git-lfs

  1. Clone the repository separately from the Isaac Lab installation (i.e. outside the IsaacLab directory)

  2. Using a python interpreter that has Isaac Lab installed, install the library

python -m pip install -e source/basic_locomotion_isaaclab

  1. If you want to play with Morphologycal Symmetries, install the repo morphosymm-rl

  2. If you want to play with Adversarial Motion Priors, install the repo amp-rsl-rl from the AMI research lab.

Run a train/play in IsaacLab

  • To train:
python scripts/rsl_rl/train.py --task=Locomotion-Aliengo-Flat --num_envs=4096 --headless
python scripts/rsl_rl/train.py --task=Locomotion-Aliengo-Rough-Blind --num_envs=4096 --headless
  • To test the policy, you can press:
python scripts/rsl_rl/play.py --task=Locomotion-Aliengo-Flat --num_envs=16
python scripts/rsl_rl/play.py --task=Locomotion-Aliengo-Rough-Blind --num_envs=16

Use AMP, Morphological Symmetries, DAGGER or Depth to Heightmap

Each of these modules has a specific README in its own script folder.

Run Hyperparameter Search

echo "import ray; ray.init(); import time; [time.sleep(10) for _ in iter(int, 1)]" | python3 (TERMINAL 1)
python3 ../basic_locomotion_isaaclab/exts/basic_locomotion_isaaclab/basic_locomotion_isaaclab/hyperparameter_tuning/tuner.py --run_mode local --cfg_file ../basic_locomotion_isaaclab/exts/basic_locomotion_isaaclab/basic_locomotion_isaaclab/hyperparameter_tuning/locomotion_aliengo_cfg.py --cfg_class LocomotionAliengoFlatTuner (TERMINAL 2)