This repository was archived by the owner on Jun 3, 2025. It is now read-only.
pytorch sparse quantized transfer learning notebook #81
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
bfineran
requested review from
jeanniefinks,
kevinaer,
markurtz,
mgoin and
natuan
|
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
This was referenced Feb 23, 2021
markurtz
reviewed
Feb 23, 2021
mgoin
reviewed
Feb 23, 2021
bfineran
force-pushed
the
qat-transfer-nb
branch
from
3e5028f to
b0a7a33
Compare
bfineran
force-pushed
the
qat-transfer-nb
branch
from
77db6d8 to
f71dda0
Compare
bfineran
force-pushed
the
qat-transfer-nb
branch
from
f71dda0 to
1cdb6fe
Compare
markurtz
approved these changes
Feb 24, 2021
mgoin
approved these changes
Feb 24, 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>
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
4 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.
example notebook for pruned quantized transfer learning of resnet50 from ImageNet to Imagenette
also included