Struct LotteryTicketPrunerBuilder 

pub struct LotteryTicketPrunerBuilder { /* private fields */ }

Builder for LotteryTicketPruner.

Implementations

impl LotteryTicketPrunerBuilder

pub fn new() -> Self

Create a new builder.

pub fn target_sparsity(self, sparsity: f32) -> Self

Set target sparsity (0.0-1.0).

pub fn pruning_rounds(self, rounds: usize) -> Self

Set number of pruning rounds.

pub fn rewind_strategy(self, strategy: RewindStrategy) -> Self

Set rewind strategy.

pub fn global_pruning(self, global: bool) -> Self

Enable global pruning across all layers.

pub fn build(self) -> LotteryTicketPruner

Build the pruner.

Trait Implementations

impl Clone for LotteryTicketPrunerBuilder

fn clone(&self) -> LotteryTicketPrunerBuilder

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LotteryTicketPrunerBuilder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for LotteryTicketPrunerBuilder

fn default() -> LotteryTicketPrunerBuilder

Returns the “default value” for a type.