Fix (#1195)

tutorials/models/1_gnn/1_gcn.py

@@ -108,8 +108,8 @@ def load_cora_data():
     data = citegrh.load_cora()
     features = th.FloatTensor(data.features)
     labels = th.LongTensor(data.labels)
-    train_mask = th.ByteTensor(data.train_mask)
-    test_mask = th.ByteTensor(data.test_mask)
+    train_mask = th.BoolTensor(data.train_mask)
+    test_mask = th.BoolTensor(data.test_mask)
     g = data.graph
     # add self loop
     g.remove_edges_from(nx.selfloop_edges(g))

tutorials/models/1_gnn/9_gat.py

@@ -276,7 +276,7 @@ def load_cora_data():
     data = citegrh.load_cora()
     features = torch.FloatTensor(data.features)
     labels = torch.LongTensor(data.labels)
-    mask = torch.ByteTensor(data.train_mask)
+    mask = torch.BoolTensor(data.train_mask)
     g = data.graph
     # add self loop
     g.remove_edges_from(nx.selfloop_edges(g))