In case all three columns are blank, don't die horribly

stanfordnlp · Nov 7, 2023 · 26db2d1 · 26db2d1
1 parent 1f34626
commit 26db2d1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stanza/models/pos/trainer.py b/stanza/models/pos/trainer.py
@@ -50,6 +50,9 @@ def update(self, batch, eval=False):
             self.model.train()
             self.optimizer.zero_grad()
         loss, _ = self.model(word, word_mask, wordchars, wordchars_mask, upos, xpos, ufeats, pretrained, word_orig_idx, sentlens, wordlens, text)
+        if loss == 0.0:
+            return loss
+
         loss_val = loss.data.item()
         if eval:
             return loss_val