starling.training.vae_train.setup_data_module

setup_data_module(cfg, effective_batch_size=None)[source]

Set up the data module for VAE training.