/baselines-experiments

Infra setup to run reinforcement learning baselines in a cloud deployment.

Primary LanguageShellMIT LicenseMIT

baselines-experiments

Infra setup to run reinforcement learning baselines in a cloud deployment.

Steps

  1. Create GCP instances to run the experiments
bash ./scripts/gcp-create-instances.sh
  1. Set up ansible inventory file. Example for GCP below
plugin: gcp_compute
projects:
  - projectname
auth_kind: serviceaccount
groups:
  rl: true
filters:
  - 'labels.algo = ppo'
  - any filters you may want
service_account_file: /home/yourdir/gcp/serviceaccount.json
  1. Install ansible galaxy roles
ansible-galaxy install geerlingguy.docker geerlingguy.pip

  1. Change ansible vault file to contain proper MongoDB login information for your database

  2. Set up experiment environment on all machines

bash ./scripts/ansible-full-setup.sh
  1. Start running the experiments
bash ./scripts/ansible-run-experiments.sh