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)