To set up the environment, use the provided scripts in the docker/ folder to build and run the codebase inside a docker container.

Use

./docker/build.sh

to build the container and

./docker/run.sh

to launch a docker container.

Then run lsvae/train_lsvae.py to start training.