|
57 | 57 | # loading data from files |
58 | 58 | #################################### |
59 | 59 | if loadFromTxt: |
60 | | - # path = osp.join("/home/yuke/.graphs/orig", args.dataset) |
61 | 60 | path = osp.join(args.dataDir, args.dataset) |
62 | 61 | dataset = custom_dataset(path, args.dim, args.classes, load_from_txt=True, verbose=verbose_mode) |
63 | 62 | else: |
@@ -149,16 +148,7 @@ def __init__(self): |
149 | 148 |
|
150 | 149 | def forward(self): |
151 | 150 | 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)) |
157 | 151 | 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)) |
162 | 152 | x = self.conv2(x, inputInfo.set_hidden()) |
163 | 153 | return F.log_softmax(x, dim=1) |
164 | 154 | else: |
|
0 commit comments