Reputation: 1
I'm trying to implement a simple two-layer network in Android NN API, but during compilation I get an error:
E/CpuExecutor: Invalid type or dimensions for model output SOFTMAX failed.
E/CpuExecutor: Invalid type or dimensions for model output SOFTMAX failed.
maybe who knows what this error is connected with? P/S/ If anyone has a working example of implementation in NN API in repo ? would be very grateful
Upvotes: 0
Views: 55
Reputation: 1
Problem solved, the size of the output tensor in the previous layer did not match. If anyone has an example of a built model for classifying an image in NNAPI I will be grateful.
Upvotes: 0