diff --git a/main.py b/main.py
index e5fe7f6..ed74963 100644
--- a/main.py
+++ b/main.py
@@ -667,6 +667,7 @@ if __name__ == "__main__":
                 collate_fn=TrainDataset.collate_fn
             ))
         for step, batch in dataloader:
+            print(batch.shape)
             sub, rel, obj, label, neg_ent, sub_samp = model.read_batch(
                 batch, 'train')