RegNet-Search-PyTorch icon indicating copy to clipboard operation
RegNet-Search-PyTorch copied to clipboard
verified publisher icon zhanghang1989

design ShuffleNetV2 search space

Open pawopawo opened this issue 5 years ago • 3 comments

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?

pawopawo avatar May 30 '20 09:05 pawopawo

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

zhanghang1989 avatar May 30 '20 16:05 zhanghang1989

Thank you for your reply,

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

pawopawo avatar May 31 '20 06:05 pawopawo

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

zhanghang1989 avatar May 31 '20 17:05 zhanghang1989