1118c16d3852beec9c0b18804d43a1fa01ef6786,examples/tgn.py,,,#,13

Before Change

                                   time_dim=100),
    aggregator_module=LastAggregator(),
    embedding_module=IdentityEmbedding(memory_dim=100))
model = model.to(device)
optimizer = torch.optim.Adam(model.parameters(), lr=0.0001)
criterion = torch.nn.BCEWithLogitsLoss()

After Change

model = TGN(
    data.num_nodes, data.x.size(-1), memory_dim=100, time_dim=100,
    message_module=IdentityMessage(raw_msg_dim=data.x.size(-1), memory_dim=100,
                                   time_dim=100),
    aggregator_module=LastAggregator()).to(device)

Instances