train_utils
This script contains the utility functions for training the LVAE model. These functions are mainly used in train.py
script.
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.
Source code in src/careamics/lvae_training/train_utils.py
get_new_model_version(model_dir)
#
A model will have multiple runs. Each run will have a different version.