starling.training.vae_train

Functions

get_checkpoint_path	Determine the checkpoint path to resume training if available.
save_config	Save the configuration to a YAML file.
setup_checkpoints	Set up model checkpoint callbacks.
setup_data_module	Set up the data module for VAE training.
setup_directories	Create necessary directories and save the configuration file.
setup_vae_model	Set up the VAE model, with support for resuming from checkpoint or fine-tuning with custom args.
wandb_init