sparseml

sparseml copied to clipboard

[SparseZoo v2 Bridge] Save transformers training artifacts to ModelDirectory directory

2

Allows creating the `ModelDirectory` local directory from the direct output of the transformers training procedure. To test: 1. You need to install locally the sparsezoo's `feature/ci-cd-refactor/validations` branch 2. Then run:...

dbogunowicz

Allows creating the `ModelDirectory` local directory from the direct output of the yolo training procedure. To test: 1. You need to install locally the sparsezoo's `feature/ci-cd-refactor/validations` branch 2. Then run:...

dbogunowicz

Bumps [mistune](https://github.com/lepture/mistune) from 0.8.4 to 2.0.3. Release notes Sourced from mistune's releases. Version 2.0.2 Fix escape_url via lepture/mistune#295 Version 2.0.1 Fix XSS for image link syntax. Version 2.0.0 First release...

dependabot[bot]

added my github username to the maintainers list

1

Chibukach

Implement label smoothing for CrossEntropy Loss

2

CrossEntropy losses (Categorical) now can take label smoothing are as parameter. In theory, ready to be reviewed, but for clarity, I will open this PR once the PyTorch 1.10.2 upgrade...

dbogunowicz

PR for refreshing content on root readme. Citations have been updated, when we have a fix for the custom dataset fine-tuning will adjust the example python.

InquestGeronimo

Hello, I was doing transfer learning on yolov5l pruned+quant model using ddp multi gpu setup. It almost completed 49/51 epochs and stopped with the following error. I was using weights...

mugdha-t

Adding git clone to Dockerfile & details about packages

2

Following commands were tested, meaning all commands in README are tested: - `docker build --build-arg CUDA_VERSION=11.1 -t sparseml_docker .` - `docker build --build-arg CUDA_VERSION=11.1 --build-arg GIT_CHECKOUT=main -t sparseml_docker .` -...

corey-nm

From community slack https://discuss-neuralmagic.slack.com/team/U03P31PUJGM: [Trunov Pavel](https://app.slack.com/team/U03P31PUJGM) [11:05 AM](https://discuss-neuralmagic.slack.com/archives/C020FPF3MQX/p1657728332430099) Hello! I'm trying to prune yolov3-tiny using sparseml. There is no recipe for tiny but I've figured out that it's possible to...

markurtz

Structured pruning for YOLOv5

8

I have been using sparseml for pruning YOLOv5 recently and I can see big improvement in inference time, however, the model size stays the same. I have realised that it...

hawrot