2bf236792fee65bd2ceab922451a230a3a986cf6,memcnn/models/tests/test_revop.py,,test_normal_vs_revblock,#Any#Any#Any#,192

Before Change

        loss2 = torch.mean(Y)

        // define the reversible function without custom backprop and define graph for model 2
        XX = Variable(X.clone().data, requires_grad=True)
        x1, x2 = torch.chunk(XX, 2, dim=1)
        if coupling == "additive":
            y1 = x1 + c1.forward(x2)
            y2 = x2 + c2.forward(y1)

After Change

    
    for seed in range(10):
        set_seeds(seed)

        X = torch.rand(2, 4, 5, 5)

        // define models and their copies

Instances