SparseML v1.4.3 Patch Release
jeanniefinks
released this
23 Mar 16:35
·
2 commits
to release/1.4
since this release
This is a patch release for 1.4.0 that contains the following changes:
- The auto install for transformers was failing due to the cutover from sklearn to scikit-learn package naming intermittently; this is no longer failing. (#1294)
- Python sparsification loggers now on instantiation they print out the directory. (#1432)
- ONNX models in YOLOv5 were improperly exported for some shapes; more shapes are now supported for dynamic models. (#1442)
- SparseML YOLOv5 validation commands were creating folders under the "zoo:" name for the SparseZoo stub; folders are now created under their ids.
- Image classificaiton training script no longer fails if optional dependency tensorboard is not installed. (#1456)
- Torchvision sparsification script now properly overrides final layer of torchvision native models. (#1455)