-
Notifications
You must be signed in to change notification settings - Fork 27
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Resnet50 for Imagenet #20
Comments
Probably because some filters that did not work during the observe process were cut out by mistake. The easiest solution is to increase OBSERVE_TIMES in prune/universal.py line 13. The definition of resnet50 is in models/imagenet/resnet50.py. Our code to prune resnet50 can be download from here -> resnet50.zip. But due to the change of code base, the code may require some modifications to run. We also provide the output of the GBN-50 experiment. |
Thankyou |
Hello, |
Hello,
I am getting an error on mismatched Batch normalization layer when pruning resnet50 for imagenet dataset.
Could you please provide Resnet50 model definition and how it is used for imagenet dataset pruning?
The text was updated successfully, but these errors were encountered: