dmx.compressor.sparse.WeightSparseMixin

class dmx.compressor.sparse.WeightSparseMixin(*args, **kwargs)

Mixin for weight-sparse modules.

__init__(*args, **kwargs) → None

Methods

__init__(*args, **kwargs)
check_sparseness_dim_consistency()
init_sparsifier()

Attributes

effective_weight
weight_sparseness