utils module

utils.utils.save_diresa(model: diresa_torch.arch.models.Diresa, path: str)

Saves Diresa model parameters to disk

Parameters:

• model – Diresa model to save

• path – path/file to save model paramaters

utils.utils.load_diresa(model: diresa_torch.arch.models.Diresa, path: str) → diresa_torch.arch.models.Diresa

Loads Diresa model parameters from disk

Parameters:

• model – Diresa model to load

• path – path/file to load model parameters

Returns:

Diresa model