Struct DistillationOutput 

pub struct DistillationOutput {
    pub total_loss: Tensor,
    pub loss_components: HashMap<String, Tensor>,
    pub teacher_predictions: Tensor,
    pub student_predictions: Tensor,
    pub feature_losses: Option<HashMap<String, Tensor>>,
    pub attention_losses: Option<HashMap<String, Tensor>>,
}

Output from distillation loss computation

Fields

total_loss: Tensor

Total distillation loss

loss_components: HashMap<String, Tensor>

Individual loss components

teacher_predictions: Tensor

Soft predictions from teacher

student_predictions: Tensor

Predictions from student

feature_losses: Option<HashMap<String, Tensor>>

Feature matching losses (if used)

attention_losses: Option<HashMap<String, Tensor>>

Attention transfer losses (if used)

Trait Implementations

impl Clone for DistillationOutput

fn clone(&self) -> DistillationOutput

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistillationOutput

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

Formats the value using the given formatter.