search

hunto / image_classification_sota

Training ImageNet / CIFAR models with sota strategies and fancy techniques such as ViT, KD, Rep, etc.
Apache License 2.0
77 stars 13 forks source link

Hello Author, I found the edgenn package in the nas_model.py and builder.py is missing，Can you provide a solution #10

Closed xjyp closed 1 year ago

hunto commented 1 year ago

Hi @xjyp ,

edgenn is an internal extension in our group to support NAS and pruning algorithms, and it is not planned to be public now.

For conventional imagenet training and all the released algorithms in image_classification_sota, you can run the code without edgenn.

If you want to use NAS and pruning algorithms, there is an alternative toolbox MMRazor by OpenMMLab, you can try it :)

xjyp commented 1 year ago

Thank you for your kind reply,I get it