【pytorch】hub

🗓 2019年12月21日 📁 文章归类: torch

版权声明：本文作者是郭飞。转载随意，标明原文链接即可。
原文链接：https://www.guofei.site/2019/12/21/torch_hub.html

https://pytorch.org/hub/

例如

import torch
model = torch.hub.load('pytorch/vision:v0.10.0', 'deeplabv3_resnet50')
model.eval()

model = models.vgg19(pretrained=True)

# 文件下载到 hub._get_torch_home()

您的支持将鼓励我继续创作！