This commit is contained in:
thanhvc3 2024-06-19 00:12:43 +07:00
parent 7448528eec
commit 6d43b88599

View File

@ -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)