设置NPU为默认设备
我们在入口文件official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py的开头添加注册如下代码,添加完成后该迁移点即完成。
1 2 | import npu_device as npu npu.open().as_default() |
父主题: 手工迁移与训练
我们在入口文件official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py的开头添加注册如下代码,添加完成后该迁移点即完成。
1 2 | import npu_device as npu npu.open().as_default() |