oneflow.save

oneflow.save(obj: Any, path: Union[str, pathlib.Path], global_dst_rank: Optional[int] = None, save_as_external_data: bool = False) → None

Save an object to a directory.

Parameters

• obj – The object to be saved

• path (str) – The directory in which the object is saved

• global_dst_rank (int, optional) – The destination rank for saving global tensors. When specified, whole tensors will be saved by the process whose rank == global_src_rank, while other processes will not do any disk I/O.