We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9331fed commit bbba2a5Copy full SHA for bbba2a5
stanza/models/coref/model.py
@@ -516,7 +516,7 @@ def train(self, log=False):
516
517
running_c_loss += c_loss.item()
518
running_s_loss += s_loss.item()
519
- if res.zero_scores.size(0) == 0:
+ if res.zero_scores.size(0) != 0:
520
running_z_loss += z_loss.item()
521
522
# log every 100 docs
@@ -525,7 +525,7 @@ def train(self, log=False):
525
'train_c_loss': c_loss.item(),
526
'train_s_loss': s_loss.item(),
527
}
528
529
logged['train_z_loss'] = z_loss.item()
530
wandb.log(logged)
531
0 commit comments