diff --git a/README.md b/README.md index aa6efd4..9d4e4ad 100644 --- a/README.md +++ b/README.md @@ -1,14 +1,15 @@ # 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 --[x] 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 + +- [x] 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