mindspore从Hub加载模型用于推理验证
2.例如,想使用GoogleNet对CIFAR-10数据集进行分类,可以在MindSpore Hub官网上使用关键词进行搜索。页面将会返回与GoogleNet相关的所有模型。进入相关模型页面之后,获得详情页。mindspore_hub.load API用于加载预训练模型,可以实现一行代码完成模型的加载。1. 在MindSpore Hub官网上搜索感兴趣的模型。3.完成模型加载后,可以使用MindS
·
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进行推理。
昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)