TUSegCheckpoint

class torch_uncertainty.callbacks.TUSegCheckpoint(save_last=False)

Keep multiple checkpoints corresponding to the best model in terms of: Mean Intersection over Union, Expected Calibration Error, Brier-Score and Negative Log-Likelihood.

Parameters:

save_last (bool | "link", optional) – When True, saves a last.ckpt copy whenever a checkpoint file gets saved. Can be set to "link" on a local filesystem to create a symbolic link. This allows accessing the latest checkpoint in a deterministic manner. Default to False.