From 417a38d2e5a5b9d168fdfb677adef6958fc82fbe Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 19 Jun 2024 00:08:27 +0700
Subject: [PATCH] test

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

diff --git a/data_loader.py b/data_loader.py
index c41a3ed..5eb7d73 100644
--- a/data_loader.py
+++ b/data_loader.py
@@ -51,6 +51,7 @@ class TrainDataset(Dataset):
     @staticmethod
     def collate_fn(data):
         triple = torch.stack([_[0] for _ in data], dim=0)
+        print(triple)
         trp_label = torch.stack([_[1] for _ in data], dim=0)
 
         if not data[0][2] is None:							# one_to_x