benchmarks.hota.train_learned.save_checkpoint
def save_checkpoint(propagator: Propagator, fuser: Fuser, path: str | Path = DEFAULT_CKPT) -> PathWrite the flattened ``{propagator,fuser}`` state to a safetensors file.
Source: unitrack/benchmarks/hota/train_learned.py:163