From 46d2468695a85dfcc2be0524caa912edefcf2391 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 16 May 2023 23:35:11 +0200 Subject: [PATCH] Update `ConvNextV2ModelIntegrationTest::test_inference_image_classification_head` (#23402) fix Co-authored-by: ydshieh --- tests/models/convnextv2/test_modeling_convnextv2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/convnextv2/test_modeling_convnextv2.py b/tests/models/convnextv2/test_modeling_convnextv2.py index 85ce5b1813..c3f8804f1c 100644 --- a/tests/models/convnextv2/test_modeling_convnextv2.py +++ b/tests/models/convnextv2/test_modeling_convnextv2.py @@ -353,5 +353,5 @@ class ConvNextV2ModelIntegrationTest(unittest.TestCase): expected_shape = torch.Size((1, 1000)) self.assertEqual(outputs.logits.shape, expected_shape) - expected_slice = torch.tensor([-0.3083, -0.3040, -0.4344]).to(torch_device) + expected_slice = torch.tensor([0.9996, 0.1966, -0.4386]).to(torch_device) self.assertTrue(torch.allclose(outputs.logits[0, :3], expected_slice, atol=1e-4))