MindSpore计算框架提供了一个官方版本的预训练模型存储库,或者叫做官方版本的预训练模型中心库,那就是 MindSpore / hub 。
首先我们需要明确概念:
第一个就是 mindspore_hub 是一个Python库,或者说是一个Python包(package),
我们需要在Python中调用mindspore_hub库中的函数,来下载官方发布的预训练模型。
官方软件地址(mindspore_hub库地址):
https://gitee.com/mindspore/hub
mindspore_hub 库的安装并不依赖计算框架MindSpore, 我们完全可以把mindspore_hub 单独看做是一个下载工具,只不过mindspore_hub是专门给计算框架MindSpore下载预训练模型的。
====================================================
MindSpore_Hub 的安装
其安装方式有多种,不过这里还是建议进行源码安装,官方给出了具体步骤:
-
从Gitee下载源码。
git clone https://gitee.com/mindspore/hub.git
-
编译安装MindSpore Hub。
cd hub python setup.py install
安装截图:
===================================================================