batch

Author: Jemoka

stanza/models/coref/model.py

@@ -575,8 +575,8 @@ def train(self, log=False):
                 c_loss = self._coref_criterion(res.coref_scores, res.coref_y)
 
                 if (res.zeros_y == 1).any():
-                    zeros_preds = res.zeros_scores[res.zeros_y != -100].reshape(-1)
-                    labels = res.zeros_y[res.zeros_y != -100].reshape(-1)
+                    zeros_preds = res.zeros_scores[res.zeros_y != -100].reshape(1, -1)
+                    labels = res.zeros_y[res.zeros_y != -100].reshape(1, -1)
                     # reweight such that the zeros and nonzeros count for equal weighting
                     # that is, artifically balance the "number of samples" by weighting between
                     # them equally