How to train MCUNet from Scratch (and perform search with TinyNAS)

mit-han-lab / mcunet

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

https://mcunet.mit.edu

MIT License

460 stars 82 forks source link

How to train MCUNet from Scratch (and perform search with TinyNAS) #26

Open zyzkevin opened 1 year ago

zyzkevin commented 1 year ago

As commented in the title, I'm having trouble running the TinyNAS search code, and training the searched network. I think all I see in the TinyNAS folder are code for the model structure but not the NAS code (presumably it is in a similar fashion to ProxylessNAS). Any help would be appreciated

h-tye commented 3 months ago

@zyzkevin I am having a similar issue. Were you ever able to make some headway in this?

zyzkevin commented 2 months ago

@zyzkevin I am having a similar issue. Were you ever able to make some headway in this?

I think I ended up using AlphaNAS's training code to train the Supernet, and then adapted the code base to MCUNet's search settings to set up the search space.