Fixed id displacement in automatic graph

Author: mancinimassimiliano

Diff for: models/graph_method.py

@@ -85,7 +85,7 @@ def get_compositional_embeddings(embeddings, pairs):
             composition_embeds = []
             for (attr, obj) in pairs:
                 attr_embed = embeddings[self.attr_to_idx[attr]]
-                obj_embed = embeddings[self.obj_to_idx[obj]]
+                obj_embed = embeddings[self.obj_to_idx[obj]+self.num_attrs]
                 composed_embed = (attr_embed + obj_embed) / 2
                 composition_embeds.append(composed_embed)
             composition_embeds = torch.stack(composition_embeds)
@@ -207,4 +207,4 @@ def forward(self, x):
         else:
             with torch.no_grad():
                 loss, pred = self.val_forward(x)
-        return loss, pred
+        return loss, pred