pylibwholegraph.torch.create_wholememory_optimizer

pylibwholegraph.torch.create_wholememory_optimizer(

embeddings: 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