test

data_loader.py

@@ -28,6 +28,8 @@ class TrainDataset(Dataset):
 
     def __getitem__(self, idx):
         ele = self.triples[idx]
+        if (idx == 0):
+            print(ele)
         triple, label, sub_samp = torch.LongTensor(ele['triple']), np.int32(
             ele['label']), np.float32(ele['sub_samp'])
         trp_label = self.get_label(label)