-
Notifications
You must be signed in to change notification settings - Fork 620
Tutorial : How to add a game to MuZero
Brendan Ortmann edited this page Oct 8, 2020
·
8 revisions
You just need to create a your-game-name.py file in the folder "games" with the configuration and the game class. You should start from an example like cartpole.
The MuZeroConfig
class holds the parameters about the game, the self-play and the training.
The Game
class is a wrapper through which the algorithm interacts with the desired game. It follows the gym API.
We have added explanations about the parameters directly in the example files.
In the code, all actions are represented by a number. It's the action_space
parameter. The legal actions are returned by the game as a list with the numbers of the actions authorized for each move.
It is recommended to adapt num_actors
, self_play_delay
, training_delay
to match the power of your computer.