88edea32e8526e13c01668aa3b00843d0ced6288,tests/estimators/classification/test_deeplearning_common.py,,test_loss_gradient_amp,#Any#Any#Any#Any#Any#,514

Before Change

    // Second test of gradients
    sub_gradients = gradients[0, 0, 14, :]
    print("sub_gradients2", sub_gradients)
    np.testing.assert_array_almost_equal(
        sub_gradients, expected_gradients_2[0], decimal=expected_gradients_2[1],
    )

After Change

    gradients = classifier.loss_gradient_framework(
        torch.tensor(x_test_mnist).to(classifier.device), torch.tensor(y_test_mnist).to(classifier.device)
    )
    gradients = gradients.cpu().numpy()

    // Test shape
    assert gradients.shape == (x_test_mnist.shape[0],) + mnist_shape

    // First test of gradients
    sub_gradients = gradients[0, 0, :, 14]

    np.testing.assert_array_almost_equal(
        sub_gradients, expected_gradients_1[0], decimal=expected_gradients_1[1],
    )

Instances