fix bugs

Author: Zhang-Each

Diff for: models/DisenMoE.py

@@ -309,9 +309,9 @@ def forward(self, batch_inputs):
 
     def get_batch_embeddings(self, batch_inputs):
         head = batch_inputs[:, 0]
-        _, disen_str = self.structure_moe(self.entity_embeddings(head))
-        _, disen_img = self.visual_moe(self.img_entity_embeddings(head))
-        _, disen_txt = self.text_moe(self.txt_entity_embeddings(head))
+        _, disen_str, _ = self.structure_moe(self.entity_embeddings(head))
+        _, disen_img, _ = self.visual_moe(self.img_entity_embeddings(head))
+        _, disen_txt, _ = self.text_moe(self.txt_entity_embeddings(head))
         return [disen_str, disen_img, disen_txt]
 
 

Diff for: models/RelMoE.py

@@ -5,8 +5,6 @@
 import torch
 from tqdm import tqdm
 
-from models.GAT import GAT
-from models.DisenMoE import DisKoMoE
 from models.RelMoE import RelMoE
 from models.model import *
 from models.modules import MIEstimator