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

fixed potentially buggy typo #405

Merged
merged 1 commit into from
Jul 24, 2024
Merged

Conversation

Alejandro-Casanova
Copy link
Contributor

the name of the function "update_regularizer" was different from that of the base MetaPruner class, so it wasn't properly overwritten. Also, in the benchmark example "reproduce/main.py" such function is never called, even though in README.md it's indicated that to apply sparse training, it should be called at the beginning of each epoch. Is this a mistake in the code? Or is it really not necessary to make such function call?

@VainF
Copy link
Owner

VainF commented Jul 24, 2024

Hello @Alejandro-Casanova, the update_regularizer is a newly added method for iterative pruning. It is indeed missing in the main.py but will not cause any error. We will fix it soon!

Thank you so much!

@VainF VainF merged commit 7da5141 into VainF:master Jul 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants