Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

fix SimulatedAnnealingPruner export mask issue #2736

Merged
merged 1 commit into from
Jul 31, 2020

Conversation

suiguoxin
Copy link
Member

self.modules_wrapper is not properly updated, this cause an error when 'export_model' is called

@QuanluZhang
Copy link
Contributor

self.modules_wrapper is not properly updated, this cause an error when 'export_model' is called

could you describe the error?

@ultmaster ultmaster requested review from colorjam and chicm-ms July 27, 2020 02:46
@suiguoxin
Copy link
Member Author

self.modules_wrapper is not properly updated, this cause an error when 'export_model' is called

could you describe the error?

A all-one mask will be exported because self.modules_wrapper is not updated, self.modules_wrapper is needed in the function export_model().

Pruner.compress() works just fine before this fix because self.bound_model is correctly updated.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants