FileSystemWriter

class tensorplay.distributed.checkpoint.FileSystemWriter(path: str | PathLike[str], single_file_per_rank: bool = True, sync_files: bool = True, thread_count: int = 1, per_thread_copy_ahead: int = 10000000, cache_staged_state_dict: bool = False, overwrite: bool = True, _extensions: Sequence[StreamTransformExtension] | None = None, serialization_format: SerializationFormat = SerializationFormat.TORCH_SAVE)[source]