commit log print

inference/harry_main.txt

@@ -3,9 +3,9 @@ that they were perfectly normal, thank you very much. They were the last
 people you'd expect to be involved in anything strange or mysterious,
 because they just didn't hold with such nonsense.
 
-Mr. Dursley was the director of a firm called Grunnings, which made
+{{Mr. Dursley}} was the director of a firm called Grunnings, which made
 drills. He was a big, beefy man with hardly any neck, although he did
-have a very large mustache. Mrs. Dursley was thin and blonde and had
+have a very large mustache. {{Mrs. Dursley}} was thin and blonde and had
 nearly twice the usual amount of neck, which came in very useful as she
 spent so much of her time craning over garden fences, spying on the
 neighbors. The Dursleys had a small son called Dudley and in their

model/memory/base_memory.py

@@ -300,7 +300,7 @@ class BaseMemory(nn.Module):
         )
 
         # Use a dummy score of 0 for froming a new cluster
-        print("Threshold: ", self.config.thresh)
+        # print("Threshold: ", self.config.thresh)
         coref_new_score = torch.cat(
             ([coref_score, torch.tensor([self.config.thresh], device=self.device)]),
             dim=0,

model/memory/entity_memory.py

@@ -140,6 +140,7 @@ class EntityMemory(BaseMemory):
                 len(mention_emb_list) % batch_size != 0
             )
             for i in range(num_batches):
+                print("Batch Number: ", i)
                 start_idx = i * batch_size
                 end_idx = min((i + 1) * batch_size, len(mention_emb_list))