test

2024-06-19 00:12:43 +07:00 · 2024-06-19 00:12:43 +07:00 · 6d43b88599
commit 6d43b88599
parent 7448528eec
1 changed files with 1 additions and 0 deletions
--- 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)