This commit is contained in:
ydshieh 2025-07-01 19:45:38 +02:00
parent 8fc6ad769b
commit 8353c2a799

View File

@ -780,4 +780,4 @@ class DeformableDetrModelIntegrationTests(unittest.TestCase):
[-10.0665, -5.8470, -7.7003],
]
)
assert torch.allclose(cpu_outputs.logits[0, :3, :3], expected_logits, atol=1e-4)
assert torch.allclose(cpu_outputs.logits[0, :3, :3], expected_logits, atol=2e-4)