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

SparseML v0.2.0

Choose a tag to compare

View all tags
@jeanniefinks jeanniefinks released this 31 Mar 23:11
v0.2.0
8890701
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

New Features:

  • Keras sparsification beta supporting pruning and examples with Keras Applications available.
  • Training and sparsification integrated with the rwightman/pytorch-image-models repository.
  • Training, sparsification, and deployment integrated with the ultralytics/yolov5 repository.
  • Integrations with the SparseZoo to run PyTorch and Keras code implemented with recipes directly from the zoo.
  • PyTorch sparse-quantized transfer learning notebook available.
  • Keras pruned ResNet models implemented.
  • Groups of modifiers enabled in SparseML recipes.

Changes:

  • Examples directory renamed to integrations.

Resolved Issues:

  • GroupedPruningMaskCreator now able to save to PyTorch state dicts.
  • Quantization-aware training compatibility issues addressed with PyTorch.
  • Docs and readme corrections made for minor issues and broken links.
  • Makefile no longer deletes files for docs compilation and cleaning.

Known Issues:

  • None
Assets 3
Loading