torchwrench.serialization.torch module¶

torchwrench.serialization.torch.dump_torch(obj: object, f: str | PathLike | BinaryIO | IO[bytes] | None = None, pickle_module: Any = <module 'pickle' from '/home/docs/.asdf/installs/python/3.12.10/lib/python3.12/pickle.py'>, pickle_protocol: int = 2, _use_new_zipfile_serialization: bool = True, _disable_byteorder_record: bool = False, *, overwrite: bool = True, make_parents: bool = True) → bytes[source]¶

torchwrench.serialization.torch.load_torch(f: str | PathLike | BinaryIO | IO[bytes], map_location: Callable[[Storage, str], Storage] | device | str | dict[str, str] | None = None, pickle_module: Any = None, *, weights_only: bool = Ellipsis, mmap: bool | None = None, **pickle_load_args: Any) → Any[source]¶