In this repository I plan to benchmark VAE variants focusing on reproducibility and ease of reading. Goals:
- One file per model
- Run all experiments with one command
- Easy experimentation
-
Install the NVIDIA drivers.
-
Install Docker ce.
-
Install nvidia-docker.
-
Clone or copy this repository, assuming it was copied in $WORKDIR
-
Run the installer
cd $WORKDIR/
sudo chmod +x install/install.sh
./install/install.sh
cd $WORKDIR/
docker run --rm --gpus all -v $PWD:/home/vae/project/ vae:latest python3 benchmark.py