设置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() |