Skip to content

Commit bbba2a5

Browse files
committed
patch: inconsistent logic for when to log zeros loss
1 parent 9331fed commit bbba2a5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

stanza/models/coref/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -516,7 +516,7 @@ def train(self, log=False):
516516

517517
running_c_loss += c_loss.item()
518518
running_s_loss += s_loss.item()
519-
if res.zero_scores.size(0) == 0:
519+
if res.zero_scores.size(0) != 0:
520520
running_z_loss += z_loss.item()
521521

522522
# log every 100 docs
@@ -525,7 +525,7 @@ def train(self, log=False):
525525
'train_c_loss': c_loss.item(),
526526
'train_s_loss': s_loss.item(),
527527
}
528-
if res.zero_scores.size(0) == 0:
528+
if res.zero_scores.size(0) != 0:
529529
logged['train_z_loss'] = z_loss.item()
530530
wandb.log(logged)
531531

0 commit comments

Comments
 (0)