def process_one_batch(self, model: MultiRelationEmbedder,
batch_edges: EdgeList) -> Stats:
if self.overrides is not None:
with override_model(model, **self.overrides):
return self._process_one_batch(model, batch_edges)
else:
return self._process_one_batch(model, batch_edges)
def process_one_batch(self, model: MultiRelationEmbedder,
batch_edges: EdgeList) -> Stats:
with override_model(
model,
num_batch_negs=self.override_num_batch_negs,
num_uniform_negs=self.override_num_uniform_negs,
):
return super().process_one_batch(model, batch_edges)
