dmx.compressor.sparse.SparsificationManager

class dmx.compressor.sparse.SparsificationManager(sparsify_modules, **kwargs)

This is a sparsification management class Similar to Scheduler that manages parameter optimization through scheduler.step(), one can call sparsification_manager.step() to update underlying score.

__init__(sparsify_modules, **kwargs)

Methods

__init__(sparsify_modules, **kwargs)
step(**kwargs)