mit-han-lab / tinyengine

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning; [NeurIPS 2022] MCUNetV3: On-Device Training Under 256KB Memory
https://mcunet.mit.edu
MIT License
792 stars 130 forks source link

how to train the quantized model on custom dataset #85

Open tonyzhangnxp opened 1 year ago

tonyzhangnxp commented 1 year ago

Hello

I 'm an engineer from NXP. I'm very interested in your incredible work. I'd read your paper "On-Device Training Under 256KB Memory", we want to reproduce your works on our platforms. but I can't find where to training the quantized model on custom dataset. I checked the mucnet repo, it only contains the model zoo, I can't find the training scripts I also checked the once-for-all repo, but it seems not support tiny subnet yet.

BR Tony