Enum CheckpointStrategy 

pub enum CheckpointStrategy {
    Every,
    EveryNth(usize),
    Sqrt,
    None,
}

Strategy for selecting which layers to checkpoint.

Variants

Every

Checkpoint every layer.

EveryNth(usize)

Checkpoint every N-th layer (layers 0, N, 2N, …).

Sqrt

Checkpoint approximately sqrt(N) layers, evenly spaced.

None

No checkpointing at all.

Implementations

impl CheckpointStrategy

pub fn select_layers(&self, total_layers: usize) -> Vec<usize>

Given total_layers, return sorted indices of layers to checkpoint.

Trait Implementations

impl Clone for CheckpointStrategy

fn clone(&self) -> CheckpointStrategy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CheckpointStrategy

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

Formats the value using the given formatter.

impl PartialEq for CheckpointStrategy

fn eq(&self, other: &CheckpointStrategy) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for CheckpointStrategy

impl StructuralPartialEq for CheckpointStrategy