dmlc--dgl
a7e941c379
* Add sparse embedding for dgl and update rgcn example * upd * Fix * Revert "Fix" This reverts commit 4da87cdfb8b8c3506b7fc7376cd2385ba8045c2a. * 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>
16 行
361 B
Python
16 行
361 B
Python
"""dgl optims."""
|
|
import importlib
|
|
import sys
|
|
import os
|
|
|
|
from ..backend import backend_name
|
|
from ..utils import expand_as_pair
|
|
|
|
def _load_backend(mod_name):
|
|
mod = importlib.import_module('.%s' % mod_name, __name__)
|
|
thismod = sys.modules[__name__]
|
|
for api, obj in mod.__dict__.items():
|
|
setattr(thismod, api, obj)
|
|
|
|
_load_backend(backend_name)
|