- Make sure your version of Brax is up to date (https://github.com/google/brax)
- Run
OMP_NUM_THREADS=1 XLA_PYTHON_CLIENT_PREALLOCATE=false python ppo.py
!! Not benchmarked !!
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.
ppo_brax_transformer
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||