world-models-pytorch

My attempt to implement "World Models" by David Ha and Jurgen Schmidhuber Link: https://arxiv.org/pdf/1803.10122.pdf

The work is in progress

TODOs

• Implement MDN-RNN
• Implement VAE
• Training for VAE
  • Random rollouts to create a dataset
  • Training function
• Training for MDN-RNN
• Evolution strategy for training Controller