benchmark_vae

Script to benchmark VAE architecture in isolation.

Usage

python -m sdfest.vae.scripts.benchmark_vae --config initialization/configs/vae_models/mug.yaml --device cuda