Enum DistillationLoss 

pub enum DistillationLoss {
    KLDivergence,
    MSE,
    CrossEntropy,
    Weighted {
        distill_weight: u8,
        ground_truth_weight: u8,
    },
}

Distillation loss function

Variants

KLDivergence

Kullback-Leibler divergence

MSE

Mean squared error

CrossEntropy

Cross-entropy

Weighted

Custom weighted combination

Fields

distill_weight: u8

Weight for distillation loss

ground_truth_weight: u8

Weight for ground truth loss

Trait Implementations

impl Clone for DistillationLoss

fn clone(&self) -> DistillationLoss

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistillationLoss

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

Formats the value using the given formatter.

impl Default for DistillationLoss

fn default() -> DistillationLoss

Returns the “default value” for a type.

impl PartialEq for DistillationLoss

fn eq(&self, other: &DistillationLoss) -> 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 DistillationLoss

impl Eq for DistillationLoss

impl StructuralPartialEq for DistillationLoss