"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ae3cbbcaf68a677f4954a7ff00b543dac03ad108"
Unverified Commit 46d24686 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Update `ConvNextV2ModelIntegrationTest::test_inference_image_classification_head` (#23402)



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent ca3df9f0
......@@ -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))
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment