Quantization aware training for executorch deployment on ethos-U

[Steph71]

Hi,

I am banging my head against this since a wile with only partly success.

Is there any example, that shows how to

• use torch.ao for quantization aware training of a simple, but not trivial NN (a cnn that classifies an MNSIT dataset should be good enough)
• deploy this NN on arm ethosU using executorch, while preserving the quantization parameters determined during QAT

A notebook, that goes through the individual steps, including some lines of explanation, what is going on would be highly appreciated.

Even better: Some information how to inspect the intermediate results for correctness and some debugging hints ...

Best Regards
Steph