class_docstr: This loss allows to distill a SwAV model
class_docstr_tok: ["This","loss","allows","to","distill","a","SwAV","model"]
code: def __init__(
      self,
      teacher_temperature: float,
      student_temperature: float,
      student_num_crops: int,
      teacher_num_crops: int,
      swapped_assignment: bool,
  ):
      super().__init__()
      self.teacher_temperature = teacher_temperature
      self.student_temperature = student_temperature
      self.student_num_crops = student_num_crops
      self.teacher_num_crops = teacher_num_crops
      self.swapped_assignment = swapped_assignment
  
file_path: vissl/vissl/losses/dino_distillation_loss.py
imports: import pprint
  from typing import List, Optional, Union
  import torch
  import torch.nn.functional
  from classy_vision.losses import ClassyLoss, register_loss
  from torch import nn
  from vissl.config import AttrDict
label_desc: Use image node_id 1 to create a new DINODistillationCriterion object from inherited base classes: nn with example: obj = DINODistillationCriterion(teacher_temperature, student_temperature, student_num_crops, teacher_num_crops, swapped_assignment)