dgl/docs/source/api/python/dgl.optim.rst at e2f33fd5cc4e3c722a65aa4e6fc79e889b740ebb - dgl - Gitea: Git with a cup of tea

pub_soft/dgl

mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00

Files

xiang song(charlie.song) a7e941c379 [Feature] Add support for sparse embedding (#2451 )

* Add sparse embedding for dgl and update rgcn example

* upd

* Fix

* Revert "Fix"

This reverts commit 4da87cdfb8.

* Fix

* upd

* upd

* Fix

* Add unitest and update impl

* fix

* Clean up rgcn example code

* upd

* upd

* update

* Fix

* update score

* sparse for sage

* remove model sparse

* upd

* upd

* remove global norm

* revert delete model_sparse.py

* update according to comments

* Fix doc

* upd

* Fix test

* upd

* lint

* lint

* lint

* upd

* upd

* clean up

Co-authored-by: Ubuntu <ubuntu@ip-172-31-56-220.ec2.internal>

2021-01-28 00:26:49 +08:00

13 lines

206 B

ReStructuredText

Raw Blame History

 .. _apioptim:
 dgl.optim
 =========
 .. automodule:: dgl.optim
 Node embedding optimizer
 -------------------------
 .. currentmodule:: dgl.optim.pytorch
 .. autoclass:: SparseAdagrad
 .. autoclass:: SparseAdam