bd346cc6ec7a75f269f3178f7c7708ed48940814,examples/tgn.py,,train,#,90

Before Change

        // Update memory and neighbor loader with ground-truth state.
        memory.update_state(src, pos_dst, t, msg)
        neighbor_loader.insert(src, pos_dst, t, msg)
        neighbor_loader.insert(pos_dst, src, t, msg)

        loss.backward()
        optimizer.step()
        memory.detach()

After Change

        // Get updated memory of all nodes involved in the computation.
        z, last_update = memory(n_id)
        z = gnn(z, last_update, edge_index, data.t[e_id], data.msg[e_id])

        pos_out = link_pred(z[assoc[src]], z[assoc[pos_dst]])
        neg_out = link_pred(z[assoc[src]], z[assoc[neg_dst]])

Instances