pub struct GradientPruner { /* private fields */ }Expand description
Gradient-based pruning (requires gradient information)
Implementations§
Source§impl GradientPruner
impl GradientPruner
Sourcepub fn new(config: PruningConfig) -> CoreResult<Self>
pub fn new(config: PruningConfig) -> CoreResult<Self>
Create a new gradient-based pruner
Sourcepub fn accumulate_gradient(&mut self, name: &str, gradient: &Array2<f32>)
pub fn accumulate_gradient(&mut self, name: &str, gradient: &Array2<f32>)
Accumulate gradient for a layer
Sourcepub fn prune_with_gradients(
&mut self,
name: &str,
weights: &Array2<f32>,
) -> CoreResult<PruningMask>
pub fn prune_with_gradients( &mut self, name: &str, weights: &Array2<f32>, ) -> CoreResult<PruningMask>
Prune using accumulated gradients
Auto Trait Implementations§
impl Freeze for GradientPruner
impl RefUnwindSafe for GradientPruner
impl Send for GradientPruner
impl Sync for GradientPruner
impl Unpin for GradientPruner
impl UnwindSafe for GradientPruner
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more