torch.save

torch.save(obj, f, pickle_module=pickle, pickle_protocol=2, _use_new_zipfile_serialization=True) [исходный код]

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

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

Дополнительные инструменты для работы с контрольной точкой см. в разделе Управление компоновкой.

Параметры:

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

Примечание

В 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)