Struct GradientPruner 

pub struct GradientPruner { /* private fields */ }

Gradient-based pruning (prune weights with smallest gradients).

Implementations

impl GradientPruner

pub fn new(config: PruningConfig) -> Self

Create a new gradient-based pruner.

pub fn update_gradients(&mut self, layer_name: &str, gradients: Array2<f64>)

Update gradient history for a layer.

pub fn prune_with_history( &self, weights: &Array2<f64>, layer_name: &str, ) -> TrainResult<(Array2<f64>, PruningMask)>

Prune based on gradient history.

Trait Implementations

impl Pruner for GradientPruner

fn prune( &self, weights: &Array2<f64>, ) -> TrainResult<(Array2<f64>, PruningMask)>

Prune weights and return pruned weights and mask.

fn generate_mask(&self, weights: &Array2<f64>) -> TrainResult<PruningMask>

Generate pruning mask without modifying weights.

fn apply_mask( &self, weights: &Array2<f64>, mask: &PruningMask, ) -> TrainResult<Array2<f64>>

Apply existing mask to weights.

fn config(&self) -> &PruningConfig

Get pruning configuration.

fn update_ratio(&mut self, iteration: usize)

Update pruning ratio for iterative pruning.