3735f4b48f52f7703944f36284b9e9ee3d1e8e5f,examples/tgn.py,,train,#,37

Before Change

        pos_out, neg_out = model(src, pos_dst, neg_dst, t, x)
        loss += criterion(pos_out, torch.ones_like(pos_out))
        loss += criterion(neg_out, torch.zeros_like(neg_out))

        total_loss += float(loss) * batch.num_events

        if (i + 1) % backprop_every == 0:
            loss = loss / backprop_every
            loss.backward()
            optimizer.step()
            model.detach_memory()

After Change

        model(src, pos_dst, neg_dst, t, x)

        if i == 2:
            raise NotImplementedError
        // pos_out, neg_out = model(src, pos_dst, neg_dst, t, x)
        // loss += criterion(pos_out, torch.ones_like(pos_out))
        // loss += criterion(neg_out, torch.zeros_like(neg_out))

Instances