unscrew with the reshaping

Author: Jemoka

stanza/models/coref/model.py

@@ -576,11 +576,12 @@ def train(self, log=False):
 
                 if (res.zeros_y == 1).any():
                     zeros_preds = res.zeros_scores[res.zeros_y != -100].reshape(-1, 2)
-                    labels = res.zeros_y[res.zeros_y != -100].reshape(-1, 2)
+                    labels = res.zeros_y[res.zeros_y != -100].reshape(-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
-                    zeros_loss = self._zeros_loss(zeros_preds, labels.float())
+                    # print(labels.float())
+                    zeros_loss = self._zeros_loss(zeros_preds, labels.long())
                 else:
                     zeros_loss = 0.0 # don't apply loss if there's nothing to learn