text: import torch
  import torch.nn.functional as F
  from train_utils import ce_loss
  
  
  class Get_Scalar:
      def __init__(self, value):
          self.value = value
  
      def get_value(self, iter):
          return self.value
  
      def __call__(self, iter):
          return self.value
  
  
  def consistency_loss(logits_w1, logits_w2):
      logits_w2 = logits_w2.detach()
      assert logits_w1.size() == logits_w2.size()
      return F.mse_loss(torch.softmax(logits_w1, dim=-1), torch.softmax(logits_w2, dim=-1), reduction='mean')
  