saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/torch/ao
History
macandro96 524d181267 [ao][sparsity] Implemented state_dict() and load_state_dict() functions (#79883) 
The state_dict() of the DataScheduler contains all the class attributes of
the scheduler other than the data_sparsifier.
Remember to store and restore the state_dict() of the data_sparsifier along
with the scheduler

Test Plan:
```python test/test_ao_sparsity.py TestBaseDataScheduler```

Differential Revision: [D37358607](https://our.internmc.facebook.com/intern/diff/D37358607)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79883
Approved by: https://github.com/HDCharles, https://github.com/z-a-f
2022-06-24 16:55:06 +00:00
..
nn [lint] upgrade mypy to latest version 2022-05-03 20:51:34 +00:00
ns ns for fx: further fixes for kwargs-only 2022-06-08 15:29:48 +00:00
quantization [ao] Adds framework for InputWeightEqualization Detector (#79916) 2022-06-24 14:51:15 +00:00
sparsity [ao][sparsity] Implemented state_dict() and load_state_dict() functions (#79883) 2022-06-24 16:55:06 +00:00
__init__.py