调优准备

样例获取

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

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

模型评估

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

网络迁移

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

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