coreai_opt.pruning.spec.ChannelStructured¶

class coreai_opt.pruning.spec.ChannelStructured[source]¶

Bases: PruningScheme

Channel-structured pruning scheme.

Entire channels (slices along axis) are pruned or kept together. Channel importance is determined by the pruning algorithm (e.g. L1 norm of each channel for magnitude-based pruning).