mindspore_hub.load API用于加载预训练模型,可以实现一行代码完成模型的加载。主要的模型加载流程如下:
-
用于推理验证
1. 在MindSpore Hub官网上搜索感兴趣的模型。
-
2.例如,想使用GoogleNet对CIFAR-10数据集进行分类,可以在MindSpore Hub官网上使用关键词进行搜索。页面将会返回与GoogleNet相关的所有模型。进入相关模型页面之后,获得详情页。GoogleNeturl
使用完成模型的加载,示例代码如下:
import mindspore_hub as mshub
import mindspore
from mindspore import context, Tensor, nn, Model
from mindspore import dtype as mstype
import mindspore.dataset.vision.py_transforms as py_transforms
context.set_context(mode=context.GRAPH_MODE,
device_target="Ascend",
device_id=0)
model = "mindspore/ascend/0.7/googlenet_v1_cifar10"
# Initialize the number of classes based on the pre-trained model.
network = mshub.load(model, num_classes=10)
network.set_train(False)
-
3.完成模型加载后,可以使用MindSpore进行推理。