本样例基于PyTorch官网提供的Imagenet数据集训练模型进行适配昇腾910 AI处理器的迁移改造。ShuffleNet模型参考PyTorch官网模型ShuffleNet V2。
模型评估主要关注算子适配情况,使用dump op方法获取ShuffleNet网络算子信息,与《支持Caffe&TensorFlow&ONNX&PyTorch算子清单》进行对比。若是发现某个算子当前暂不支持,对于简单场景我们可以考虑先暂时替换成类似的算子,或者把该算子单独放到CPU上执行两种方式规避;复杂场景不支持算子需要参见《算子开发指南》进行算子适配开发。