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

Before Change

    model.conv2.reset_parameters()

    torch.cuda.synchronize()
    t = time.perf_counter()
    for _ in range(0, 200):
        train()
    torch.cuda.synchronize()
    t = time.perf_counter() - t
    acc.append(test())
    print("Run:", run, "Test Accuracy:", acc[-1], "Time:", t)

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

After Change

    for _ in range(0, 200):
        train()
        val = test(val_mask)
        if val > old_val:
            old_val = val
            cur_test = test(test_mask)

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

acc = torch.FloatTensor(acc)

Instances