Skip to content

Train LVAE

Source

This script is meant to load data, initialize the model, and provide the logic for training it.

get_mean_std_dict_for_model(config, train_dset)

Computes the mean and std for the model. This will be subsequently passed to the model.

get_new_model_version(model_dir)

A model will have multiple runs. Each run will have a different version.