design ShuffleNetV2 search space

zhanghang1989 / RegNet-Search-PyTorch

Search for RegNet using PyTorch

MIT License

316 stars 44 forks source link

design ShuffleNetV2 search space #8

Open pawopawo opened 4 years ago

pawopawo commented 4 years ago

Thanks for your excellent work! Are the search criteria in regnet suitable for ShuffleNetV2? If I want to search a shuffleNet model with FLOPs 60M, can I design a search space based on this code?

zhanghang1989 commented 4 years ago

Yes, the code is easy to extend. You can create another file for shuffleNet the the folder https://github.com/zhanghang1989/RegNet-Search-PyTorch/tree/master/arch

pawopawo commented 4 years ago

Thank you for your reply,

Do different blocks (shufflenet or ResNeSt) and different FLOPs require manual adjustment of the range of these hyper-parameters?

zhanghang1989 commented 4 years ago

The default ranges are quite large. It's better to adjust them according to the FLOPs to improve the sampling efficiency.