test

2024-06-19 00:16:05 +07:00 · 2024-06-19 00:16:05 +07:00 · a064d12763
commit a064d12763
parent 6d43b88599
1 changed files with 1 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -102,6 +102,7 @@ class Main(object):

        self.p.num_ent = len(self.ent2id)
        self.p.num_rel = len(self.rel2id) // 2
+        print("Num rel: " + str(self.p.num_rel))
        self.p.embed_dim = self.p.k_w * \
            self.p.k_h if self.p.embed_dim is None else self.p.embed_dim