调优准备

样例获取

本样例基于PyTorch官网提供的Imagenet数据集训练模型进行适配昇腾910 AI处理器的迁移改造。ShuffleNet模型参考PyTorch官网模型ShuffleNet V2

ShuffleNet为PyTorch内置模型,了解更多内置模型请前往PyTorch官网

模型评估

模型评估主要关注算子适配情况,使用dump op方法获取ShuffleNet网络算子信息,与支持Caffe&TensorFlow&ONNX&PyTorch算子清单进行对比。若是发现某个算子当前暂不支持,对于简单场景我们可以考虑先暂时替换成类似的算子,或者把该算子单独放到CPU上执行两种方式规避;复杂场景不支持算子需要参见算子开发指南进行算子适配开发。

网络迁移

训练脚本迁移请参见模型迁移与训练

完成迁移后,在拉起脚本训练的命令中指定参数arch为shufflenet_v2_x1_0。