wassname/world-models-pytorch

mirror of https://github.com/wassname/world-models-pytorch.git synced 2026-08-11 11:28:32 +08:00

T

Junhong XuandGitHub 8fc6145242 Update README.md

2018-04-04 22:49:39 -04:00

.gitignore

added rnn

2018-04-03 23:48:51 -04:00

controller.py

added rnn

2018-04-03 23:48:51 -04:00

model.py

added rnn

2018-04-03 23:48:51 -04:00

README.md

Update README.md

2018-04-04 22:49:39 -04:00

rnn.py

finished MDN-RNN and need to implement VAE

2018-04-04 22:43:48 -04:00

train.py

added rnn

2018-04-03 23:48:51 -04:00

vae.py

finished MDN-RNN and need to implement VAE

2018-04-04 22:43:48 -04:00

README.md

world-models-pytorch

My attempt to implement "World Models" by David Ha and Jurgen Schmidhuber

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