From 26db2d1c65d41a18e941a40a64c97167be887505 Mon Sep 17 00:00:00 2001 From: John Bauer Date: Mon, 6 Nov 2023 23:46:26 -0800 Subject: [PATCH] In case all three columns are blank, don't die horribly --- stanza/models/pos/trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/stanza/models/pos/trainer.py b/stanza/models/pos/trainer.py index f8d848db24..16f0ef9db6 100644 --- 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