World Models applied to the Open AI Sonic Retro Contest
A student implementation of the World Models paper with documentation.
Ongoing project.
Once you've done that, you will need to train the 3 components :
python train_vae.py
python train_lstm.py --folder=xxx
python train_controller.py --folder=xxx
where xxx is the folder number created in saved_models/
While training the VAE and the LSTM, pictures will be saved in a folder results/
Once you're done, you can use your best trained controller to play a random level using :
python play_best --folder=xxx
Dont forget to change the RENDER_TICK in const.py to 1, so you can see what's happening.