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.