graphstorm icon indicating copy to clipboard operation
graphstorm copied to clipboard

Published 20 hours ago • verified publisher icon awslabs

[Feature request] Save optimizer states of learnable sparse embedding.

Open classicsong opened this issue 1 year ago • 0 comments

Currently, GraphStorm does not save/load optimizer states of learnable sparse embeddings. (https://github.com/awslabs/graphstorm/blob/main/python/graphstorm/model/utils.py#L336-L337). We need to support it.

classicsong avatar Apr 10 '23 06:04 classicsong