From 03f42561c6623862ca5e611fabc4de230e7d7432 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 19 Jun 2024 00:05:57 +0700
Subject: [PATCH] test

---
 data_loader.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/data_loader.py b/data_loader.py
index 03e0af5..c41a3ed 100644
--- a/data_loader.py
+++ b/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)