Skip to content

GitLab

Unverified Commit c2b8af45 authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

change the backend to mxnet. (#894)

parent 3e454476
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
tutorials/models/5_giant_graph/1_sampling_mx.py
View file @ c2b8af45
...@@ -57,6 +57,9 @@ from dgl.data import RedditDataset ...@@ -57,6 +57,9 @@ from dgl.data import RedditDataset
import mxnet as mx import mxnet as mx
from mxnet import gluon from mxnet import gluon
# Load MXNet as backend
dgl.load_backend('mxnet')
# load dataset # load dataset
data = RedditDataset(self_loop=True) data = RedditDataset(self_loop=True)
train_nid = mx.nd.array(np.nonzero(data.train_mask)[0]).astype(np.int64) train_nid = mx.nd.array(np.nonzero(data.train_mask)[0]).astype(np.int64)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment