e2db3b3f1d3d23cd5bc1e295835e0f4b33e95447,examples/cora_gcn.py,,,#,57

Before Change

    print("Run:", run, "Test Accuracy:", acc[-1])

acc = torch.FloatTensor(acc)
print("Mean:", acc.mean(), "Stddev:", acc.std())

After Change

        return F.log_softmax(x, dim=1)


model = Net()
if torch.cuda.is_available():
    train_mask, val_mask = train_mask.cuda(), val_mask.cuda()
    test_mask, model = test_mask.cuda(), model.cuda()

optimizer = torch.optim.Adam(model.parameters(), lr=0.01, weight_decay=0.005)


def train():

Instances