Skip to content
Snippets Groups Projects

This repository allows to run Rail Environment multi agent training with the RLLib Library.

It should be clone inside the main flatland repository.

Installation:

pip install ray

pip install gin-config

To start a grid search on some parameters, you can create a folder containing a config.gin file (see example in grid_search_configs/n_agents_grid_search/config.gin.

Then, you can modify the config.gin file path at the end of the grid_search_train.py file.

The results will be stored inside the folder, and the learning curves can be visualized in tensorboard:

tensorboard --logdir=/path/to/foler_containing_config_gin_file.