oneflow.save¶
-
oneflow.
save
(obj: Any, path_or_buffer: Union[os.PathLike, BinaryIO, IO[bytes], 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_or_buffer – a file-like object (has to implement write and flush) or a string or os.PathLike object containing a file name
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.
save_as_external_data (bool) – useful only if path_or_buffer is a string or os.PathLike object containing a file name