From 0de3762dcee7b0ad483893642dd24d36447d3605 Mon Sep 17 00:00:00 2001 From: tqchen Date: Wed, 30 Oct 2019 15:40:56 -0700 Subject: [PATCH 1/2] [CI] Move gpu docker binary to cuda10 --- Jenkinsfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Jenkinsfile b/Jenkinsfile index ceb5771d82f4..59f9cbb61eb1 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -45,7 +45,7 @@ // ci_lint = "tvmai/ci-lint:v0.51" -ci_gpu = "tvmai/ci-gpu:v0.54" +ci_gpu = "tvmai/ci-gpu:v0.55" ci_cpu = "tvmai/ci-cpu:v0.54" ci_i386 = "tvmai/ci-i386:v0.52" From e80ea08a9bfb80aa57a1cdf56eee164b1a608a4d Mon Sep 17 00:00:00 2001 From: tqchen Date: Wed, 30 Oct 2019 22:07:08 -0700 Subject: [PATCH 2/2] Fix the gcn tutorial --- tutorials/frontend/build_gcn.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tutorials/frontend/build_gcn.py b/tutorials/frontend/build_gcn.py index e97c83c6c44e..077b066f7f62 100644 --- a/tutorials/frontend/build_gcn.py +++ b/tutorials/frontend/build_gcn.py @@ -46,6 +46,7 @@ import torch.nn as nn import torch.nn.functional as F import dgl +import networkx as nx from dgl.nn.pytorch import GraphConv class GCN(nn.Module): @@ -88,7 +89,7 @@ def load_dataset(dataset="cora"): # Remove self-loops to avoid duplicate passing of a node's feature to itself g = data.graph - g.remove_edges_from(g.selfloop_edges()) + g.remove_edges_from(nx.selfloop_edges(g)) g.add_edges_from(zip(g.nodes, g.nodes)) return g, data @@ -110,7 +111,7 @@ def evaluate(data, logits): Parameters ---------- dataset: str - Name of dataset. You can choose from ['cora', 'citeseer', 'pubmed']. + Name of dataset. You can choose from ['cora', 'citeseer', 'pubmed']. num_layer: int number of hidden layers @@ -251,7 +252,7 @@ def GraphConv(layer_name, ###################################################################### # Prepare the parameters needed in the GraphConv layers # ------------------ -# +# import numpy as np import networkx as nx