don't init embeddings in projection embedder on loading from checkpoint
@@ -29,6 +29,7 @@ class ProjectionEmbedder(KgeEmbedder):
self.dropout = self.get_option("dropout")
self.regularize = self.check_option("regularize", ["", "lp"])
self.projection = torch.nn.Linear(self.base_embedder.dim, self.dim, bias=False)
+ if not init_for_load_only:
self._init_embeddings(self.projection.weight.data)
def _embed(self, embeddings):
