This repository re-implements the core architectural components and training code from the paper CITRIS. We re-use most of the evaluation and dataset utility code from the official implementation.
python train_vae.py --data_dir $LINK_TO_PONG_YOUR_PONG_DATASET