How ntnu-arl/olympus_lab works
Olympus-Lab provides Isaac Lab reinforcement-learning environments for quadruped walking, horizontal and vertical jumping, and in-flight attitude control. The main workflow is a training or playback invocation selecting a task, which creates a task-specific simulation environment; the RL policy supplies actions and receives observations and rewards as the simulated Olympus robot evolves. Task environments also handle initialization, curricula, and task state where applicable. The README documents Earth, Mars, and zero-gravity variants plus policy playback; paths and call links for some of these capabilities are only supported at directory level because their implementation was not sampled.
Main entry points: jump_env.py Jump environment · vertical_jump_env.py Vertical jump environment · walk_env.py Walking environment · play.py Play policy
Components 23
Policy workflows
- Train policy
train.py - Play policy
play.py
Locomotion tasks
- Walking environment
envs/walk/walk_env.py - Jump environment
envs/jump/jump_env.py - Vertical jump environment
envs/vertical_jump/vertical_jump_env.py - Attitude environment
envs/attitude_control/attitude_control_env.py - Jump playback setup
envs/jump/jump_play_env.py - Vertical playback setup
envs/vertical_jump/vertical_jump_play_env.py - Mars task variants
envs
Robot and control
- Olympus articulation
robot/olympus.py - CubeMars actuators
robot/cube_mars/cubmars_actuator_model.py - Motor command filter
control/motor_command_filter.py - Kinematics
kinematics
Task support
- Jump initializers
envs/jump/initalization - Vertical initializers
envs/vertical_jump/initalization - Jump state machine
envs/jump/jump_state_machine.py - Vertical state machine
envs/vertical_jump/jump_state_machine.py - Jump curricula
envs/jump/curriculum.py - Vertical curriculum
envs/vertical_jump/curriculum.py - Simulation logging
logger/logger.py
Other components
- Task user
- Isaac Lab simulator
- RL Games
Connections 38
- Task user → Train policy: selects task
- Task user → Play policy: selects task
- Train policy → RL Games: trains policy
- RL Games → Walking environment: steps environment
- RL Games → Jump environment: steps environment
- RL Games → Vertical jump environment: steps environment
- RL Games → Attitude environment: steps environment
- Play policy → Walking environment: runs task
- Play policy → Jump playback setup: runs task
- Play policy → Vertical playback setup: runs task
- Play policy → Attitude environment: runs task
- Jump playback setup → Jump environment: configures playback
- Vertical playback setup → Vertical jump environment: configures playback
- Walking environment → Isaac Lab simulator: simulates
- Jump environment → Isaac Lab simulator: simulates
- Vertical jump environment → Isaac Lab simulator: simulates
- Attitude environment → Isaac Lab simulator: simulates
- Walking environment → Olympus articulation: controls articulation
- Jump environment → Olympus articulation: controls articulation
- Vertical jump environment → Olympus articulation: controls articulation
- Attitude environment → Olympus articulation: controls articulation
- Olympus articulation → CubeMars actuators: uses actuators
- Walking environment → Motor command filter: filters commands
- Jump environment → Motor command filter: filters commands
- Walking environment → Kinematics: uses calculations
- Jump environment → Jump initializers: initializes episodes
- Vertical jump environment → Vertical initializers: initializes episodes
- Jump environment → Jump state machine: updates state
- Vertical jump environment → Vertical state machine: updates state
- Jump environment → Jump curricula: uses curriculum
- Vertical jump environment → Vertical curriculum: uses curriculum
- Walking environment → Simulation logging: logs episodes
- Jump environment → Simulation logging: logs episodes
- Vertical jump environment → Simulation logging: logs episodes
- Attitude environment → Simulation logging: logs episodes
- Mars task variants → Walking environment: provides Mars tasks
- Mars task variants → Jump environment: provides Mars tasks
- Mars task variants → Vertical jump environment: provides Mars tasks