一、Torch Hub概述
Pytorch Hub 是一个预训练模型存储库,旨在促进研究的可重复性。Torch Hub 允许您发布预先训练的模型,以帮助促进研究共享和可重复性。
Torch Hub 在其官方展示中总共纳入了48个研究模型(目前为止)。音频模型8个、生成式模型2个、自然语言处理 (NLP)3个、可编写脚本20个、视觉模型等等。这些模型还在基准数据集(例如Kinetics 400和COCO 2017)上进行了训练。
在您的项目中使用这些模型很容易,使用torch.hub.load功能就可以调用。让我们看一个例子来说明它是如何工作的。
二、示例:使用DCGAN生成图像
首先进行模型的加载,如果下载失败,可以看失败的log手动下载,手动下载之后。修改source='local',默认是'github'。
# USAGE
# python inference.py
# import the necessary packages
import matplot