Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "050ef1451682a64a70394f372021572c59dc4c28"
Unverified Commit f03580fb authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Fix DETR integration test (#12734)

parent f42d9dcc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_detr.py
View file @ f03580fb
......@@ -524,4 +524,4 @@ class DetrModelIntegrationTests(unittest.TestCase):
expected_slice_masks = torch.tensor(
[[-7.7558, -10.8788, -11.9797], [-11.8881, -16.4329, -17.7451], [-14.7316, -19.7383, -20.3004]]
).to(torch_device)
self.assertTrue(torch.allclose(outputs.pred_masks[0, 0, :3, :3], expected_slice_masks, atol=1e-4))
self.assertTrue(torch.allclose(outputs.pred_masks[0, 0, :3, :3], expected_slice_masks, atol=1e-3))
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