diff --git a/main.py b/main.py index 43caac5..670705a 100644 --- 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