"left" and "right" not in gragh
hello-big-world opened this issue · 0 comments
hello-big-world commented
for k1 in ["left", "right"]:
graph[k1] = dict()
for k2 in ["u", "v"]:
temp = [graphs[i][k1][k2] + counts[i] for i in range(batch_size)]
temp = [
x if x.dim() > 0 else graph["pre"][0]["u"].new().resize_(0)
for x in temp
]
graph[k1][k2] = torch.cat(temp)
KeyError: 'left'