wassname/world-models-pytorch

mirror of https://github.com/wassname/world-models-pytorch.git synced 2026-06-27 13:47:21 +08:00

T

junhong 3ee35e440e Merge remote-tracking branch 'origin/master'

2018-04-12 23:22:57 -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

dataset.py

finished VAE and VAE loss function, working on dataset

2018-04-12 23:22:50 -04:00

model.py

added rnn

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

README.md

Update README.md

2018-04-04 22:50:29 -04:00

rnn.py

finished VAE and VAE loss function, working on dataset

2018-04-12 23:22:50 -04:00

rollouts.py

finished VAE and VAE loss function, working on dataset

2018-04-12 23:22:50 -04:00

train.py

added rnn

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

vae.py

finished VAE and VAE loss function, working on dataset

2018-04-12 23:22:50 -04:00

README.md

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