Remove duplicate log_softmax calculation (#158)

Removed duplicate log_softmax call on ref_outputs.

Author: NewHorse2022

src/trainer/utils.py

@@ -18,7 +18,6 @@ def compute_kl_divergence(model, target_model, inputs):
     with torch.no_grad():
         ref_outputs = target_model(**inputs)
 
-    ref_probs = F.log_softmax(ref_outputs.logits, dim=-1)
     ref_probs = F.log_softmax(ref_outputs.logits, dim=-1)
     ref_probs = ref_probs.view(-1, ref_outputs.logits.shape[-1])