try sep vit

This commit is contained in:
thanhvc3 2024-04-28 09:42:39 +07:00
parent 37b01708b4
commit 3243b1d963

View File

@ -478,7 +478,11 @@ class Main(object):
batch, 'train')
pred = self.model.forward(sub, rel, neg_ent, self.p.train_strategy)
loss = self.model.loss(pred, label, sub_samp)
try:
loss = self.model.loss(pred, label, sub_samp)
except Exception as e:
print(pred)
raise e
loss.backward()
self.optimizer.step()