mirror of
https://github.com/wassname/world-models-pytorch.git
synced 2026-06-27 18:27:21 +08:00
Junhong Xu
16 lines
393 B
Markdown
16 lines
393 B
Markdown
# 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
|