622e0e1a9ed3099994697f86ae34f83d056626d8,test/test_image.py,TestBgRToRgb,test_gradcheck,#TestBgRToRgb#,192

Before Change

    def test_gradcheck(self):

        // prepare input data
        data = torch.ones(2, 3, 1, 1)

        data = utils.tensor_to_gradcheck_var(data)  // to var

        assert gradcheck(image.BgrToRgb(), (data,),

After Change

    def test_gradcheck(self):

        // prepare input data
        data = torch.tensor([[[1., 1.],
                              [1., 1.]],

                             [[2., 2.],
                              [2., 2.]],

                             [[3., 3.],
                              [3., 3.]]])  // 3x2x2

        data = utils.tensor_to_gradcheck_var(data)  // to var

Instances