-
Notifications
You must be signed in to change notification settings - Fork 324
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
Unable to use strip_pruning for subclass model #965
Comments
AFAIK evenif you apply prune_low_magnitude to sublayers of your subclass model, the pruning scheduling logic (callback based) during the training is not working due to it can't find your pruned layer. So I think your model is not trained as expected with pruning. Unfortunately, subclass model supports is very weak in these days. Maybe there's some work around but it's still a brittle solution. @rino20 Hi rino20, do you know any recent workaround for the subclass model supports? or any recommendation? |
Hi @Xhark, thanks for your kind help. |
Hi @LIU-FAYANG
|
Hi @rino20, thanks for your help! Thanks for your kind help! |
Hi Thanks for providing the details.
|
Hi @rino20, thanks for your kind help, I'll try the method u mentioned above these days. edit: |
Hi @rino20 , I am new to the model pruning world. I would like to try to prune the mobilenetv3 small model to use it on a smaller device. could you give me the high-level steps to approve this task? appreciate any comment you may have on this. |
Hi, is there any work around method to use tfmot.sparsity.keras.strip_pruning on subclass models? I tried to use prune_low_magnitude on subclass by apply it to the layers in subclasses as a work around method. So I would like to ask is there method like this to use strip_pruning on subclasses models? Any help would be great:)
The text was updated successfully, but these errors were encountered: