2b968e05a99392d1010cec3fb40c97c592f15065,examples/pna.py,,,#,11

Before Change

                in_channels=dataset.num_node_features,
                out_channels=dataset.num_classes, avg_d=data.num_edges / data.num_nodes,
                edge_features=bool(dataset.num_edge_features)).to(device)
data = data.to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=0.01)


def train():

After Change

dataset = Planetoid(path, dataset, transform=T.NormalizeFeatures())
data = dataset[0].to(device)

D = torch.bincount(data.edge_index[1], minlength=data.num_nodes).float()

model = PNAConv(aggregators=args.aggregators.split(), scalers=args.scalers.split(), towers=args.towers,
                divide_input=args.divide_input,
                pretrans_layers=args.pretrans_layers, posttrans_layers=args.posttrans_layers,

Instances