简介
使用前必读
本章节旨在通过一个简单的实践,让开发者快速了解基于CANN进行模型迁移、训练并加载模型进行推理的大致流程。
快速入门的操作以Atlas 800 训练服务器(型号:9010)为例进行操作说明,若您使用其他型号的昇腾AI处理器,支持的功能特性及操作可能存在差异。
环境准备
功能描述
图片分类任务是最基础的计算机视觉应用,给定一张图像,可以判断其所属的类别,例如猫、狗、飞机、汽车等。
本实践样例将PyTorch框架的ResNet50分类网络训练脚本迁移到昇腾平台上执行训练,并将训练好的模型保存、转换为昇腾平台适配的离线OM模型,最终通过AscendCL应用程序加载并执行此模型,实现最基础的图片分类AI应用。
父主题: 快速入门