diff --git a/README.md b/README.md index cc11b12..aa6efd4 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,14 @@ # world-models-pytorch My attempt to implement "World Models" by David Ha and Jurgen Schmidhuber + +## 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