torch.hub.load

torch.hub.load 是 PyTorch 中的一个函数，可以方便地从 GitHub 上加载预训练模型或代码库。它的基本用法是：

import torch.hub

model = torch.hub.load(repo_or_dir, model, pretrained=True)

其中，`repo_or_dir` 是 GitHub 代码库的地址或本地路径，`model` 是要加载的模型名，`pretrained` 参数指示是否加载预训练的模型。例如，要加载 PyTorch 官方提供的 ResNet-50 模型，可以使用以下代码：

import torch.hub

model = torch.hub.load('pytorch/vision', 'resnet50', pretrained=True)

这将自动从 GitHub 上下载并加载名为 `resnet50` 的模型，并返回一个 PyTorch 模型实例。