pytorch
af4e2b2c - [ao][sparsity] Implemented the step() function (#79822)

Commit
3 years ago
[ao][sparsity] Implemented the step() function (#79822) The step() function calls the implemented get_schedule_param() and updates the data_group dictionary of the data sparsifier. Test Plan: ```python test/test_ao_sparsity.py TestBaseDataScheduler``` Differential Revision: [D37358609](https://our.internmc.facebook.com/intern/diff/D37358609) Pull Request resolved: https://github.com/pytorch/pytorch/pull/79822 Approved by: https://github.com/z-a-f
Author
macandro96
Committer
Parents
Loading