[MAIXPY]OSError:ERR_OUTPUTS_SET :kpu.set_outputs() to set shape

[Ratchapol0507]

HELLO @AIWintermuteAI
I followed your GitHub operation and finally trained a 2 class model mushroom (Edible,Poison) , then i flash the you firmware maixpy.bin which you post in the another question, after that i flash the model to the k210 , when run the script mushroom ,but get error [MAIXPY] OSError:kpu: set_outputs to set shape .

a = kpu.set_outputs(task, 0, 7,7,21) #the actual shape needs to match the last layer shape of your model(before Reshape)
a = kpu.set_outputs(task, 1, 14,14,21) #the actual shape needs to match the last layer shape of your model(before Reshape)

 !!!Could you help me fix this problem?!!!

Code colab me used to train the model, follow the link below.
https://colab.research.google.com/drive/1gVoFMYr8BOKIzy_YojohqbKYpNfRrzLw?usp=sharing

 The image below has the problem.