WorldModels

An implementation of the ideas from this paper https://arxiv.org/pdf/1803.10122.pdf for CS-GY 6613.

Building with Docker

Used Dockerfile.cpu and launch-docker-cpu.sh for all of my work

For report, please see PDF in repo.

Name		Name	Last commit message	Last commit date
Latest commit History 171 Commits
.ipynb_checkpoints		.ipynb_checkpoints
.vscode		.vscode
controller		controller
custom_envs		custom_envs
data		data
log		log
rnn		rnn
vae		vae
vaegan		vaegan
.env		.env
.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
06_train_vaegan.py		06_train_vaegan.py
Dockerfile		Dockerfile
Dockerfile.cpu		Dockerfile.cpu
Dockerfile.gpu		Dockerfile.gpu
GS_GY_6613_Project_2.pdf		GS_GY_6613_Project_2.pdf
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_02_vae.py		check_02_vae.py
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