pylibwholegraph.torch.embedding.create_wholememory_optimizer

pylibwholegraph.torch.embedding.create_wholememory_optimizer(embeddings: Union[WholeMemoryEmbedding, List[WholeMemoryEmbedding]], optimizer_type: str, param_dict: dict)

Create WholeMemoryOptimizer. :param embeddings: WholememoryEmbeddings to set the Optimizer :param optimizer_type: Type of the Optimizer :param param_dict: parameters of the optimizer :return: WholeMemoryOptimizer