Skip to content

GitLab

Unverified Commit a199b58a authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

Fix (#1195)

parent cd9599ab
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tutorials/models/1_gnn/1_gcn.py
View file @ a199b58a
...@@ -108,8 +108,8 @@ def load_cora_data(): ...@@ -108,8 +108,8 @@ def load_cora_data():
data = citegrh.load_cora() data = citegrh.load_cora()
features = th.FloatTensor(data.features) features = th.FloatTensor(data.features)
labels = th.LongTensor(data.labels) labels = th.LongTensor(data.labels)
train_mask = th.ByteTensor(data.train_mask) train_mask = th.BoolTensor(data.train_mask)
test_mask = th.ByteTensor(data.test_mask) test_mask = th.BoolTensor(data.test_mask)
g = data.graph g = data.graph
# add self loop # add self loop
g.remove_edges_from(nx.selfloop_edges(g)) g.remove_edges_from(nx.selfloop_edges(g))
......
tutorials/models/1_gnn/9_gat.py
View file @ a199b58a
...@@ -276,7 +276,7 @@ def load_cora_data(): ...@@ -276,7 +276,7 @@ def load_cora_data():
data = citegrh.load_cora() data = citegrh.load_cora()
features = torch.FloatTensor(data.features) features = torch.FloatTensor(data.features)
labels = torch.LongTensor(data.labels) labels = torch.LongTensor(data.labels)
mask = torch.ByteTensor(data.train_mask) mask = torch.BoolTensor(data.train_mask)
g = data.graph g = data.graph
# add self loop # add self loop
g.remove_edges_from(nx.selfloop_edges(g)) g.remove_edges_from(nx.selfloop_edges(g))
......
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