【pytorch】hub



2019年12月21日    Author:Guofei

文章归类: 0x26_torch    文章编号: 265

版权声明:本文作者是郭飞。转载随意,标明原文链接即可。本人邮箱
原文链接: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', pretrained=True)
model.eval()
model = models.vgg19(pretrained=True)

# 文件下载到 hub._get_torch_home()

其它

关于 TensorboardX: https://www.bilibili.com/video/BV1LP4y1F7UQ


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