dmlc--dgl
Da Zheng
5567f4a4d9
* support mxnet. * add mxnet version of GCN. * rename mxnet.nd as F. * add mxnet GAT. * enable GPU for GCN. * fix MXNet GCN train. * Use adam to optimize GAT * support more operators. * support sparse arrays. * update mxnet backend. * support index_copy. * remove NN. * update mxnet backend. * temp check in. * fix data conversion. * add test. * clean up mxnet backend. * update mxnet examples. * Revert "remove NN." This reverts commit d815d9a0ec619f9ce9099c48cd35db9d8e947483. * temp disable MXNet version of NN.
14 行
322 B
Python
14 行
322 B
Python
"""Package nn modules"""
|
|
from __future__ import absolute_import
|
|
|
|
import os
|
|
|
|
__backend__ = os.environ.get('DGLBACKEND', 'pytorch').lower()
|
|
|
|
if __backend__ == 'numpy':
|
|
pass
|
|
elif __backend__ == 'pytorch':
|
|
from .pytorch import *
|
|
elif __backend__ != 'mxnet':
|
|
raise Exception("Unsupported backend %s" % __backend__)
|