wassname/world-models-pytorch

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

T

Mike Clark dacb51da28 remove sqrt from l2_dist

2018-05-08 20:12:34 +08: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

remove sqrt from l2_dist

2018-05-08 20:12:34 +08: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