A JAX Implementation of the Soft Actor Critic Algorithm
Best run in docker:
docker build --tag=<your-tag> .
For requirement details consult environment.yml
and Dockerfile
.
python main.py pendulum
This will select the config specified in configs/pendulum.json
. Command line arguments can be used to override arguments in config.
To do...