WorldModels

An implementation of the ideas from this paper https://arxiv.org/pdf/1803.10122.pdf

For full installation and run instructions see this blog post:

Building with Docker

See Dockerfile.cpu and launch-docker-cpu.sh or equivalent GPU files for instructions.

Name		Name	Last commit message	Last commit date
.ipynb_checkpoints		.ipynb_checkpoints
controller		controller
custom_envs		custom_envs
data		data
log		log
rnn		rnn
vae		vae
.gitignore		.gitignore
01_generate_data.py		01_generate_data.py
01_generate_data.sh		01_generate_data.sh
02_train_vae.py		02_train_vae.py
03_generate_rnn_data.py		03_generate_rnn_data.py
04_train_rnn.py		04_train_rnn.py
05_train_controller.py		05_train_controller.py
Dockerfile		Dockerfile
Dockerfile.cpu		Dockerfile.cpu
Dockerfile.gpu		Dockerfile.gpu
LICENSE		LICENSE
README.md		README.md
check_01_generate_data.ipynb		check_01_generate_data.ipynb
check_02_vae.ipynb		check_02_vae.ipynb
check_04_rnn.ipynb		check_04_rnn.ipynb
check_05_controller.ipynb		check_05_controller.ipynb
check_05_controller_dream.ipynb		check_05_controller_dream.ipynb
config.py		config.py
env.py		env.py
es.py		es.py
launch-docker-cpu.sh		launch-docker-cpu.sh
launch-docker-gpu.sh		launch-docker-gpu.sh
model.py		model.py
requirements.txt		requirements.txt