When I integrated my model that I previously trained using yolov8 into Android, I saw that it always runs on cpu. What can I do to make PyTorch work on vulkan and nnapi. When I add backend=vulkan parameter to optimise_for_mobile function, Unknown custom class type vulkan.BatchNormPackedContext. Please ensure it is registered.: I get this error. I could not solve this in any way. How can I make a yolo model in PyTorch format to work with vulkan and nnapi on Android?
1 Like