Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Conversation

@bfineran
Copy link
Contributor

@bfineran bfineran commented May 6, 2021

fixes bug that defaults pytorch mask creators to be 'unstructured' in certain training flows

updates release version to 0.3.1 for this hotfix

* fix propagation of mask creator objects in PT pruning modifier

* quality - need type() to handle inheretence
@bfineran bfineran requested review from a team and dhuangnm May 6, 2021 18:08
@bfineran bfineran self-assigned this May 6, 2021
@bfineran bfineran requested review from kevinaer, markurtz and natuan and removed request for a team May 6, 2021 18:08
@dhuangnm
Copy link
Member

dhuangnm commented May 6, 2021

Can we also add a test for it so we can verify things are working?

@bfineran
Copy link
Contributor Author

bfineran commented May 6, 2021

@dhuangnm the test file is updated in this PR to check that the correct mask creator is set

@dhuangnm
Copy link
Member

dhuangnm commented May 6, 2021

@dhuangnm the test file is updated in this PR to check that the correct mask creator is set

oh ok, cool thanks.

@dhuangnm
Copy link
Member

dhuangnm commented May 6, 2021

Can we update the version.py to 0.3.1 for the cherry-pick?

@bfineran bfineran merged commit cae2a1e into release/0.3 May 11, 2021
@bfineran bfineran deleted the cherry-pick-mask-type-fix-0.3 branch May 11, 2021 21:52
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants