Setting NPU as Default Device
Add the following code lines at the beginning of the entry point file official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py.
1 2 | import npu_device as npu npu.open().as_default() |
Parent topic: Manual Porting and Training