This repository was archived by the owner on Jun 3, 2025. It is now read-only.
Serialize/deserialize MaskedLayer #69
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
markurtz
suggested changes
Feb 17, 2021
| raise NotImplementedError("Not implemented") | ||
|
|
||
|
|
||
| class SchedulerRegistry: |
There was a problem hiding this comment.
We should be able to get rid of this class entirely and avoid the potential confusion that comes with a more complicated setup. Specifically, from_config would be a class method on the PruningScheduler and an internal registry of the derived classes can be created by overriding the new definition
There was a problem hiding this comment.
Agreed that decorator is not a good solution here for registering classes. I've removed it and fixed using init_subclass.
natuan
force-pushed
the
tuan/keras_custom_serialization
branch
from
8029d49 to
83035c1
Compare
markurtz
approved these changes
Feb 18, 2021
bfineran
approved these changes
Feb 18, 2021
markurtz
added a commit
that referenced
this pull request
Feb 25, 2021
* GA code, toctree links (#61) - added tracking for docs output - added help links for docs output * Update README.md (#64) removed placeholder reference to comingsoon repo in favor of active repo * add decorator for flaky tf sparsity tests (#65) * enable modifier groups in SparseML recipes (#66) * enable modifier groups in SparseML recipes * unit tests for YAML modifier list loading * make build argument for nightly builds (#63) * rst url syntax correction (#67) correcting double slash at the end of URLs with updates to index.rst pre-compilation * match types explicitly in torch qat quant observer wrappers (#68) * docs updates (#71) enhancing left nav for Help; after this merge, the docs need to be rebuilt for this repo so docs.neuralmagic.com can be refreshed. cc @markurtz * Rename KSModifier to PruningModifier (#76) * Rename ConstantKSModifier to ConstantPruningModifier * Rename GradualKSModifier to GMPruningModifier * Fix broken link for Optimization Recipes (#75) * Serialize/deserialize MaskedLayer (#69) * Serialize/deserialize MaskedLayer * Remove unused symbols * Register pruning scheduler classes for serialization * removed ScheduledOptimizer, moved logic to ScheduledModifierManager (#77) * load recipes directly from SparseZoo (#72) * load recipes into Managers from sparsezoo stubs * moving recipe_type handling to SparseZoo only, supporting loading SparseZoo recipe objects * Revert "removed ScheduledOptimizer, moved logic to ScheduledModifierManager (#77)" (#80) This reverts commit 6073abb. * Update for 0.1.1 release (#82) * Update for 0.1.1 release - update python version to 0.1.1 - setup.py add in version parts and _VERSION_MAJOR_MINOR for more flexibility with dependencies between neural magic packages - add in deepsparse optional install pathway * missed updating version to 0.1.1 * rename examples directory to integrations (#78) * rwightman/pytorch-image-models integration (#70) * load checkpoint file based on sparsezoo recipe in pytorch_vision script (#83) * ultralytics/yolov5 integration (#73) * pytorch sparse quantized transfer learning notebook (#81) * load qat onnx models for conversion from file path (#86) * Sparsification update (#84) * Sparsification update - update sparsification descriptions and move to preferred verbage * update from comments on deepsparse for sparsification * Update README.md Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update README.md Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update README.md Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update README.md Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update docs/source/index.rst Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update docs/source/index.rst Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update docs/source/index.rst Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update docs/source/index.rst Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * Update docs/source/recipes.md Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * fix links in index.rst from reviewed content * update overviews and taglines from doc Co-authored-by: Jeannie Finks <74554921+jeanniefinks@users.noreply.github.com> * blog style readme for torch sparse-quant TL notebook (#85) Co-authored-by: Jeannie Finks (NM) <74554921+jeanniefinks@users.noreply.github.com> Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com> Co-authored-by: Eldar Kurtic <eldar.ciki@gmail.com> Co-authored-by: Tuan Nguyen <tuan@neuralmagic.com>
markurtz
pushed a commit
that referenced
this pull request
Sep 1, 2021
stray bad link to clean up for clean archive
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Support serialization and deserialization for models constructed for pruning using MaskedLayer layers.