Update modeling_dicow.py

modeling_dicow.py

@@ -332,9 +332,6 @@ class DiCoWForConditionalGeneration(DiCoWGenerationMixin, WhisperForConditionalG
             dec_loss1 = loss_fct(dec_lm_logits.view(-1, self.config.vocab_size), labels.reshape(-1))
             dec_loss2 = loss_fct(dec_lm_logits.view(-1, self.config.vocab_size), upp_labels.reshape(-1))
             dec_loss = torch.hstack((dec_loss1[..., None], dec_loss2[..., None])).min(dim=-1).values.mean()
-            if wandb.run is not None:
-                wandb.log({"dec_loss": dec_loss})
-                wandb.log({"ctc_loss": ctc_loss})
             loss = (1 - self.config.ctc_weight) * dec_loss + self.config.ctc_weight * ctc_loss