Update inference.py

inference.py

@@ -79,11 +79,14 @@ def evaluate(model, dataloader):
     criterionMCM = nn.MSELoss()
     
     with torch.no_grad():
-        for batch in dataloader:
+        for idx, batch in enumerate(dataloader):
             input_ids = batch[0]
             masked_tokens = batch[1]
             masked_pos = batch[2]
-            
+
+            if idx == 0:
+                print(input_ids[0])
+                
             logits_lm, output = model(input_ids, masked_pos)
             
             output_batch_preproc = output