torch.save

torch.save(obj, f, pickle_module=pickle, pickle_protocol=DEFAULT_PROTOCOL, _use_new_zipfile_serialization=True) [source]

Сохраняет объект в файл на диске.

См. также: Сохранение и загрузка тензоров

Параметры:

• obj (object) – сохраняемый объект
• f (Union[str, PathLike, BinaryIO, IO[bytes]]) – файл-подобный объект (должен реализовывать write и flush) или строка или объект os.PathLike, содержащий имя файла
• pickle_module (Any) – модуль, используемый для сериализации метаданных и объектов
• pickle_protocol (int) – можно указать, чтобы переопределить протокол по умолчанию

Примечание

Общая конвенция PyTorch заключается в сохранении тензоров с расширением файла .pt.

Примечание

PyTorch сохраняет совместное использование хранилища при сериализации. Подробнее см. Сохранение и загрузка тензоров сохраняет представления.

Примечание

В релизе PyTorch 1.6 произошел переход к новому формату файла на основе zipfile. torch.save Возможность загружать файлы в старом формате torch.load по-прежнему сохраняется. Если по какой-либо причине требуется torch.save использовать старый формат, передайте аргумент _use_new_zipfile_serialization=False.

Пример

>>> # Save to file
>>> x = torch.tensor([0, 1, 2, 3, 4])
>>> torch.save(x, 'tensor.pt')
>>> # Save to io.BytesIO buffer
>>> buffer = io.BytesIO()
>>> torch.save(x, buffer)