sparseml

sparseml copied to clipboard

Let modifiers implement epoch adjustment when composing recipes

3

This change simplifies the manager's recipe combination, also fixes a bug that the current code at the manager level does not account for all epoch related attributes from the quantization...

natuan

don't save epoch on IC one_shot checkpoints

1

default saved epoch for `one_shot` in the IC flows is `-1` due to `Trainer` initialization. This will cause issues on model load since the checkpoint recipe will be initialized to...

bfineran

dbogunowicz

Update YOLACT integration docs

2

This PR updates YOLACT documentation in `sparseml/integrations`, The commands have been updated to use our new training flows and the setup instructions have been made simpler(cause of the latest refactor)...

rahul-tuli

Create deployment folder for transformers integration

1

Testing: First I set up the sparsezoo model (picked a stub and downloaded it). I remove deployment directory: ```bash (sparsezoo_venv) damian@lambdaquad:~/.cache/sparsezoo/9768075d-2cf0-4bc7-98c3-fb6441a2ec36$ ls deployment model.md model.onnx recipe sample_inputs.tar.gz sample_outputs.tar.gz training (sparsezoo_venv)...

dbogunowicz

Create deployment folder for IC integration

1

## Testing ```bash python export.py --dataset imagenette --dataset-path /home/damian/imagenette --checkpoint-path pytorch_vision/mobilenet_v1_imagenette__03/training/checkpoint-best.pth ``` results in ```bash Model id is set to _imagenette__16 already downloaded imagenette ImagenetteSize.s160 2022-08-11 06:54:56 __main__ INFO exporting...

dbogunowicz

Bumps [nbconvert](https://github.com/jupyter/nbconvert) from 6.0.7 to 6.3.0. Commits cefe0bf Release 6.3.0 a534fb9 Release 6.3.0b0 87920c5 Add changelog for 6.3.0 (#1669) dd6d9c7 add slide numbering (#1654) 5d2c5e2 Update state filter (#1664) 11ea593...

dependabot[bot]

Draft: Supporting torch 1.12

2

Todos: - Guard specific torch function calls in torch version checks: - [ ] fuse_modules - [ ] onnx.export - [ ] Get all the unit tests passing (there are...

corey-nm

Question

2

Hi, I hope all is fine, Thank you so much for sharing this repository, my question is can I apply this optimization on Yolox or just this optimization for bert...

LamnouarMohamed

Hi, Is it possible to add new tasks or models? e.g. BART, text generation, seq2seq. I guess it could be if I implement a new task and pipeline, but not...

s-gerrard48