function

benchmarks.hota.train_learned.save_checkpoint

def save_checkpoint(propagator: Propagator, fuser: Fuser, path: str | Path = DEFAULT_CKPT) -> Path

Write the flattened ``{propagator,fuser}`` state to a safetensors file.

Source: unitrack/benchmarks/hota/train_learned.py:163