class_docstr: Teacher-Student knowledge distillation helper.
  Use this object when training a model with KD and a teacher model.
  
  Args:
      teacher_model (TrainableModel): teacher model
      temperature (float, optional): KD temperature. Defaults to 1.0.
      dist_w (float, optional): distillation loss weight. Defaults to 0.1.
      loss_w (float, optional): student loss weight. Defaults to 1.0.
      loss_function (str, optional): loss function to use (kl for KLDivLoss,
          mse for MSELoss)
docstr: Get teacher logits
  
  Args:
      inputs: input
  
  Returns:
      teachr logits
code: def get_teacher_logits(self, inputs):
      """
      Get teacher logits
  
      Args:
          inputs: input
  
      Returns:
          teachr logits
      """
      return self.teacher.get_logits(inputs)
  
file_path: nlp-architect/nlp_architect/nn/torch/distillation.py
imports: import argparse
  import logging
  import torch
  import torch.nn
  import torch.nn.functional
  from nlp_architect.models import TrainableModel
label_desc: Use image node_id 2 for calling the TeacherStudentDistill obj's underlying member method code with example usage: obj.get_teacher_logits(inputs) and returns: self