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进行推理。

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐