[CPU] Fix ScaledDotProductAttension accuracy problem (#21217)

This commit is contained in:
Luo Cheng authored and GitHub committed 2023-11-26 14:47:04 +04:00
1 parent 493a338ad2
commit 91660b1c05
5 files changed
+99 -44

No files matched your search

@@ -75,7 +75,8 @@ class TestTimmConvertModel(TestConvertModel):
@pytest.mark.parametrize("name", ["mobilevitv2_050.cvnets_in1k",
"poolformerv2_s12.sail_in1k",
"vit_base_patch8_224.augreg_in21k"])
"vit_base_patch8_224.augreg_in21k",
"beit_base_patch16_224.in22k_ft_in22k"])
@pytest.mark.precommit
def test_convert_model_precommit(self, name, ie_device):
self.run(name, None, ie_device)