try sep vit

2024-04-28 09:42:39 +07:00 · 2024-04-28 09:42:39 +07:00 · 3243b1d963
commit 3243b1d963
parent 37b01708b4
1 changed files with 5 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -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()