From 6d43b885999620b856a71904bcf91093fd36f277 Mon Sep 17 00:00:00 2001 From: thanhvc3 Date: Wed, 19 Jun 2024 00:12:43 +0700 Subject: [PATCH] test --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index 9909869..43caac5 100644 --- a/main.py +++ b/main.py @@ -479,6 +479,7 @@ class Main(object): sub, rel, obj, nt_rel, label, neg_ent, sub_samp = self.read_batch( batch, 'train') + print(nt_rel) pred = self.model.forward(sub, rel, nt_rel, neg_ent, self.p.train_strategy) loss = self.model.loss(pred, label, sub_samp)