pub struct LayerPruner { /* private fields */ }Expand description
Layer pruning (remove entire layers)
Implementations§
Source§impl LayerPruner
impl LayerPruner
pub fn new() -> Self
pub fn with_importance_scores(scores: HashMap<String, f32>) -> Self
Sourcepub fn analyze_model<M>(&mut self, model: &M) -> Result<()>where
M: Model,
pub fn analyze_model<M>(&mut self, model: &M) -> Result<()>where
M: Model,
Calculate layer importance using model-level metrics (simplified)
Sourcepub fn get_pruning_candidates(
&self,
config: &PruningConfig,
) -> Result<Vec<String>>
pub fn get_pruning_candidates( &self, config: &PruningConfig, ) -> Result<Vec<String>>
Get layers that would be pruned based on current importance scores
Trait Implementations§
Auto Trait Implementations§
impl Freeze for LayerPruner
impl RefUnwindSafe for LayerPruner
impl Send for LayerPruner
impl Sync for LayerPruner
impl Unpin for LayerPruner
impl UnsafeUnpin for LayerPruner
impl UnwindSafe for LayerPruner
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