设置NPU为默认设备

我们在入口文件official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py的开头添加注册如下代码,添加完成后该迁移点即完成。

import npu_device as npu
npu.open().as_default()