torch.jit.load

`torch.jit.load` is a function in PyTorch that loads a serialized TorchScript model from a file or a file-like object. It returns a `torch.jit.ScriptModule` object, which can be used to run the model.

Syntax:

torch.jit.load(filepath_or_buffer, map_location=None, **kwargs)

Parameters:

- `filepath_or_buffer` (str or file-like object) – The path to the serialized TorchScript model or a file-like object containing the serialized model.
- `map_location` (str or torch.device or callable, optional) – A string specifying the device where the model will be loaded, or a torch.device object representing the device, or a callable that takes a string parameter and returns a torch.device object. Default is None, which means the model will be loaded on the same device where it was originally saved.
- `**kwargs` – Additional keyword arguments that will be passed to `torch.load()` function.

Returns:

- A `torch.jit.ScriptModule` object representing the loaded model.

Example:

import torch

# Load the serialized model from a file
model = torch.jit.load('model.pt')

# Load the serialized model from a file-like object
with open('model.pt', 'rb') as f:
    model = torch.jit.load(f)

# Load the serialized model on a different device
device = torch.device('cpu')
model = torch.jit.load('model.pt', map_location=device)