SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update defaults and logic to match old FE (#20065)

Browse Source 
* Update defaults and logic to match old FE

* Use docker run rest values
This commit is contained in:
amyeroberts
2022-11-04 19:14:56 +00:00
committed by GitHub
parent c06d555647
commit d68c46026b
5 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/perceiver/test_modeling_perceiver.py
View File

@@ -903,7 +903,7 @@ class PerceiverModelIntegrationTest(unittest.TestCase):
expected_shape = torch.Size((1, model.config.num_labels))
self.assertEqual(logits.shape, expected_shape)
expected_slice = torch.tensor([-1.1653, -0.1993, -0.7521], device=torch_device)
expected_slice = torch.tensor([-1.1652, -0.1992, -0.7520], device=torch_device)
self.assertTrue(torch.allclose(logits[0, :3], expected_slice, atol=1e-4))