d041ccdc8112fa0d62ee9ad4ab4cecf0fd8e2cf3,tests/gradient_tests.py,TestCompareAdjointGradient,problem,#TestCompareAdjointGradient#,99

Before Change

                return torch.mm(y**3, self.A)

        y0 = torch.tensor([[2., 0.]]).to(TEST_DEVICE).requires_grad_(True)
        t_points = torch.linspace(0., 25., 10).to(TEST_DEVICE).requires_grad_(True)
        func = Odefunc().to(TEST_DEVICE)
        return func, y0, t_points

    def test_dopri5_adjoint_against_dopri5(self):

After Change

                return torch.mm(y**3, self.A)

        y0 = torch.tensor([[2., 0.]], device=TEST_DEVICE, requires_grad=True)
        t_points = torch.linspace(0., 25., 10, device=TEST_DEVICE, requires_grad=True)
        func = Odefunc().to(TEST_DEVICE)
        return func, y0, t_points

    def test_dopri5_adjoint_against_dopri5(self):

Instances