tum-ei-eda / mlonmcu

Tool for the deployment and analysis of TinyML applications on TFLM and MicroTVM backends
Apache License 2.0
28 stars 12 forks source link

Just a question... #117

Closed vladBaciu closed 6 months ago

vladBaciu commented 6 months ago

Can custom models and datasets be used easily on this project ?

I am wondering if I can evaluate a custom model on a custom dataset that is not publicly available. If yes, could you give me some hints ?

PhilippvK commented 6 months ago

The model zoo we provide is entirely optional. There are a few ways to use your own models instead

The only limitation is that your model type has to be supported by one of mlonmcus frontends.

Regarding datasets: Since mlonmcu is only working with pre-trained models, it does not really care about datasets. The only case when the data becomes relevant is when validation the model outputs against a golden reference. This feature is supported by MLonMCU (see https://github.com/tum-ei-eda/mlonmcu/blob/main/ipynb/Examples/ValidateOutputs.ipynb) but we are currently working on a more sophisticated solution for model output verification...

I hope this answers your question!