-
Mayank Mittal authored
# Description So far we have only been able to run the training script in headless mode. However, this has posed limitations in our workflows since we always want to check the policy in between the training iterations and not just at the end. This MR creates a simple UI that allows toggling between full rendering and no rendering. * In full-rendering, the environment is rendered at every `env.step()` call, i.e. it happens at the environment frequency (not the physics dt or rendering dt). * In no-rendering, the environment rendering is disabled. However, we still throttle the UI at every five environment steps to still allow UI interactions (otherwise keyboard and mouse interactions get blocked). Additionally, the MR fixes the behavior of UI interactions when hitting the pause/stop buttons: * When paused: the environment blocks itself and waits for play to happen. * When stopped: the environment closes itself and shutdowns the simulation app, instead of just throwing an error. However, there is no free cost here. Running with the GUI still causes some slowdowns. For training `Isaac-Ant-v0` with RL-Games, the following numbers are obtained: * headless: ~170,000 FPS (sim), ~98,000 FPS (total) * UI with no-rendering: ~150,000 FPS (sim), 88,000 FPS (total) * UI with full-rendering: ~40,000 FPS (sim), 25,000 FPS (total) ## Type of change - New feature (non-breaking change which adds functionality) - Breaking change (fix or feature that would cause existing functionality to not work as expected) ## Screenshots https://github.com/isaac-orbit/orbit/assets/12863862/2bc61fdc-c392-43ed-bbd5-12c743ca3150 ## Checklist - [x] I have run the [`pre-commit` checks](https://pre-commit.com/) with `./orbit.sh --format` - [ ] 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 - [x] I have updated the changelog and the corresponding version in the extension's `config/extension.toml` file
| Name |
Last commit
|
Last update |
|---|---|---|
| .github | ||
| .vscode | ||
| docker | ||
| docs | ||
| source | ||
| .dockerignore | ||
| .flake8 | ||
| .gitignore | ||
| .pre-commit-config.yaml | ||
| LICENSE | ||
| README.md | ||
| VERSION | ||
| orbit.sh | ||
| pyproject.toml |