torch.jit.load

torch.jit.load(f, map_location=None, _extra_files=None) [source]

Загрузка ScriptModule или ScriptFunction, сохранённых ранее с помощью torch.jit.save.

Все ранее сохранённые модули, независимо от устройства, сначала загружаются в ЦП, а затем перемещаются на устройства, с которых они были сохранены. Если это не удаётся (например, потому что система выполнения не имеет определённых устройств), генерируется исключение.

Параметры:

• f – объект типа файл (должен реализовывать read, readline, tell и seek) или строка, содержащая имя файла
• map_location (строка или torch.device) – Упрощённая версия map_location в torch.jit.save для динамического переназначения хранилищ на другой набор устройств.
• _extra_files (словарь, сопоставляющий имена файлов с содержимым) – Дополнительные имена файлов, заданные в map, будут загружены, а их содержимое будет сохранено в предоставленном map.

Возвращает:

Объект ScriptModule.

Пример:

import torch
import io

torch.jit.load('scriptmodule.pt')

# Load ScriptModule from io.BytesIO object
with open('scriptmodule.pt', 'rb') as f:
    buffer = io.BytesIO(f.read())

# Load all tensors to the original device
torch.jit.load(buffer)

# Load all tensors onto CPU, using a device
buffer.seek(0)
torch.jit.load(buffer, map_location=torch.device('cpu'))

# Load all tensors onto CPU, using a string
buffer.seek(0)
torch.jit.load(buffer, map_location='cpu')

# Load with extra files.
extra_files = {'foo.txt': ''}  # values will be replaced with data
torch.jit.load('scriptmodule.pt', _extra_files=extra_files)
print(extra_files['foo.txt'])