SparseML v0.1.0 First GitHub Release
Welcome to our initial release on GitHub! Older release notes can be found here.
New Features:
- Keras Alpha for optimizing models using pruning added.
- PyTorch 1.7 is supported.
- PyTorch distributed supported for built-in training flows.
- Torchvision models added to PyTorch ModelRegistry class.
- MakeFile flows and utilities implemented for GitHub repo structure.
Changes:
- Software packaging updated to reflect new GitHub distribution channel, from file naming conventions to license enforcement removal.
- Migration made to use the SparseZoo package for loading pre-trained models and recipes.
- Migration made to use DeepSparse Engine for analyzing and benchmarking ONNX models.
- ONNX and ONNXRuntime dependency versions updated to include latest.
Resolved Issues:
- Infinite recursion resolved for the PyTorch ScheduledOptimizer in nested optimizer flows.
- tf2onnx folding nodes now working with Sparsify.
Known Issues:
- TensorFlow pre-trained models are not pushed currently in the SparseZoo and will fail to load.
- TensorFlow V1 is no longer being built for newer operating systems such as Ubuntu 20.04. Therefore, SparseML with TensorFlow V1 is unsupported on these operating systems as well.