环境准备
一定要找到Pytorch和CANN的对应的版本,否则会帮忙填坑。
https://www.hiascend.com/developer/download/community/result?module=dl+pt+cann
也可以使用其他的训练框架。
https://www.hiascend.com/developer/download/
训练数据
使用数据集使用flowers102
训练数据
性能展示
ResNet18
910b-resnet18-speed.jpg
ResNet50
910b-resnet50-speed.jpg