analogdevicesinc / ai8x-training

Model Training for ADI's MAX78000 and MAX78002 Edge AI Devices
Apache License 2.0
86 stars 76 forks source link

Add NAS KWS model, Dynamic Augmentation and Automated Evaluation Notebook #280

Closed alicangok closed 5 months ago

alicangok commented 6 months ago

Major changes:


Summary of Improvements:

Along with the previous PR, we have improved the KWS20 accuracy from ~86.5% to 92.5% on the validation set which includes augmented samples, and from 87.6% to 93.7% on the clean test set.

The impact of each change on the KWS20 accuracy are as follows:

alicangok commented 6 months ago

Changed the pull request to "draft" mode: Awaiting requested changes at: https://github.com/alicangok/ai8x-training/pull/1