从MindSpore Hub加载模型总结

转载地址：https://bbs.huaweicloud.com/forum/thread-112009-1-1.html

作者：李响

邮箱：lixiang97@huawei.com

想使用GoogleNet对CIFAR-10数据集进行分类，可以在MindSpore Hub官网上使用关键词GoogleNet进行搜索。页面将会返回与GoogleNet相关的所有模型。进入相关模型页面之后，获得详情页url，相关链接：https://www.mindspore.cn/resources/hub

示例代码如下

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",      #根据自身环境设置，若使用CPU或者GPU可替换

                     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)

# ...

完成模型加载后，可以使用MindSpore进行推理，参考https://www.mindspore.cn/tutorial/inference/zh-CN/r1.1/multi_platform_inference.html。