diff --git a/combo/combo_model.py b/combo/combo_model.py
index b87eb1c0719fdd1e8fc4e1e4695559deb81644cb..3ab90c88580882a029429c0904cf4a925407c803 100644
--- a/combo/combo_model.py
+++ b/combo/combo_model.py
@@ -173,6 +173,7 @@ class ComboModel(Model, FromParameters):
                 mapped_gold_labels = []
                 for _, cat_indices in self.morphological_feat.slices.items():
                     try:
+                        print("Feats dimensions: ",  feats.shape, "cat_indices shape", len(cat_indices), "min and max", min(cat_indices), max(cat_indices))
                         mapped_gold_labels.append(feats[:, :, cat_indices].argmax(dim=-1))
                     except TypeError:
                         raise ConfigurationError('Feats is None - if no feats are provided, the morphological_feat property should be set to None.')