torch.save

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

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

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

Параметры

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

Примечание

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

Примечание

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

Примечание

В версии PyTorch 1.6 был изменен torch.save на новый формат файла, основанный на zipfile. 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)