50822819b664cd9d751d410a5eabd4cc0688311d,examples/tgn.py,,,#,38

Before Change

link_pred = LinkPredictor(in_channels=100).to(device)

optimizer = torch.optim.Adam(
    list(model.parameters()) + list(link_pred.parameters()), lr=0.0001)
criterion = torch.nn.BCEWithLogitsLoss()

assoc = torch.empty(data.num_nodes, dtype=torch.long, device=device)

After Change

sampler = NeighborSampler(data, size=10)
train_data, val_data, test_data = data.train_val_test_split(
    val_ratio=0.15, test_ratio=0.15)

Instances