7038ae40ddcb5b6f50097e4b0c33a39d2b6d3a72,tests/links_tests/readout_tests/test_set2set.py,,check_backward,#Any#Any#Any#,57

Before Change

        return readout(atom_data).data,

    gx, = gradient_check.numerical_grad(f, (atom.data, ), (y.grad, ))
    numpy.testing.assert_allclose(cuda.to_cpu(gx), cuda.to_cpu(atom.grad),
                                  atol=1e-2, rtol=1e-2)


def test_backward_cpu(readout, data):
    // type: (Set2Set, Tuple[numpy.ndarray, numpy.ndarray]) -> None

After Change

        readout.reset_state()
        return readout(atom_data),

    gradient_check.check_backward(
        f, (atom_data), y_grad, atol=1e-1, rtol=1e-1)

    // gx, = gradient_check.numerical_grad(f, (atom.data, ), (y.grad, ))
    // numpy.testing.assert_allclose(cuda.to_cpu(gx), cuda.to_cpu(atom.grad),
    //                               atol=1e-2, rtol=1e-2)

Instances