Skip to content

Commit f129823

Browse files
author
Yuke Wang
committed
update model script
1 parent b5b394e commit f129823

1 file changed

Lines changed: 0 additions & 10 deletions

File tree

GNNAdvisor/GNNA_main.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,6 @@
5757
# loading data from files
5858
####################################
5959
if loadFromTxt:
60-
# path = osp.join("/home/yuke/.graphs/orig", args.dataset)
6160
path = osp.join(args.dataDir, args.dataset)
6261
dataset = custom_dataset(path, args.dim, args.classes, load_from_txt=True, verbose=verbose_mode)
6362
else:
@@ -149,16 +148,7 @@ def __init__(self):
149148

150149
def forward(self):
151150
x = dataset.x
152-
153-
# print("[Foward]-1: {}\n{}\n{}\n{}\n{}".format(inputInfo.row_pointers, inputInfo.column_index,
154-
# inputInfo.degrees, inputInfo.partPtr, inputInfo.part2Node))
155-
# print("[Foward]-1: partSize: {}, dimWorker: {}, warpPerBlock: {}".format(inputInfo.partSize, \
156-
# inputInfo.dimWorker, inputInfo.warpPerBlock))
157151
x = F.relu(self.conv1(x, inputInfo.set_input()))
158-
# print("[Foward]-2: {}\n{}\n{}\n{}\n{}".format(inputInfo.row_pointers, inputInfo.column_index,
159-
# inputInfo.degrees, inputInfo.partPtr, inputInfo.part2Node))
160-
# print("[Foward]-2: partSize: {}, dimWorker: {}, warpPerBlock: {}".format(inputInfo.partSize, \
161-
# inputInfo.dimWorker, inputInfo.warpPerBlock))
162152
x = self.conv2(x, inputInfo.set_hidden())
163153
return F.log_softmax(x, dim=1)
164154
else:

0 commit comments

Comments
 (0)