Skip to content

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