The main requirements can be found in requirements.txt.
To install the requirements, you can follow the instructions below:
pip install -r requirements.txt
Additionally, Mujoco should be installed.
The main training loop is in learner.py, the VAE set-up and losses are in vae.py, the model design is in models/, the RL algorithms are in algorithms/, and the hyperparameters are in config/. You need to modify the hyperparameters for the specific environment you want to run.
To evaluate, run
python main.py --env-type <env_name>
which will use hyperparameters from config/args_<env_name>.py.