starling.training.vae_train.setup_vae_model

setup_vae_model(cfg)[source]

Set up the VAE model, with support for resuming from checkpoint or fine-tuning with custom args.