univi.utils.io.save_checkpoint

univi.utils.io.save_checkpoint(path, model_state=None, optimizer_state=None, extra=None, *, model=None, optimizer=None, trainer_state=None, scaler_state=None, config=None, strict_label_compat=True)