For Researchers —

Explore and extend models
from the latest
cutting edge research


How it works —

Publishing Models

PyTorch Hub支持通过添加简单的hubconf.py文件将预训练的模型(模型定义和预训练的权重)发布到GitHub存储库.

Loading models

用户可以使用torch.hub.load() API加载预先训练的模型.

这是一个示例,显示了如何从pytorch/vision库中加载resnet18入口点.

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