Skip to content
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

the size of model become larger #4

Open
zhengmiao1996 opened this issue Jul 22, 2019 · 5 comments
Open

the size of model become larger #4

zhengmiao1996 opened this issue Jul 22, 2019 · 5 comments
Assignees
Labels
bug Something isn't working

Comments

@zhengmiao1996
Copy link

As we know resnet50 is 94M, but when I load the model and retrain it for several epochs, it become 280M, I don't know why because I model.summary() the .h5 and see the parameters are same. Thank you very much.

@gaborvecsei gaborvecsei added the bug Something isn't working label Jul 28, 2019
@gaborvecsei
Copy link
Owner

What is the version of Tensorflow and Keras?

@zhengmiao1996
Copy link
Author

tensorflow 1.8 and keras 2.24

@gaborvecsei
Copy link
Owner

Then I would advice to change to the newest version of Tensorflow and try out with that. It is defined in the requirements: https://github.com/gaborvecsei/Ridurre-Network-Filter-Pruning-Keras/blob/master/requirements.txt#L1

With this version it can happen that the keras.backend.clear_session() or the tf.reset_default_graph() does not work reliably: https://github.com/gaborvecsei/Ridurre-Network-Filter-Pruning-Keras/blob/master/ridurre/base_filter_pruning.py#L160

@gaborvecsei gaborvecsei self-assigned this Jul 29, 2019
@ccsone
Copy link

ccsone commented Dec 3, 2019

@zhengmiao1996 at last it works? now i also want to have a try to prune the resnet network

@Lin-N-A-N
Copy link

Has this problem solved? I got the same situation, I have trained the FER plus datasets by your example(model_pruning_example) but use the origin resnet not yours, then I got a 280M model

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

4 participants