diff --git a/tests/models/timm_backbone/test_modeling_timm_backbone.py b/tests/models/timm_backbone/test_modeling_timm_backbone.py index 1f0b729bbf54..ea1eb18d8a93 100644 --- a/tests/models/timm_backbone/test_modeling_timm_backbone.py +++ b/tests/models/timm_backbone/test_modeling_timm_backbone.py @@ -101,7 +101,7 @@ def test_config(self): self.config_tester.run_common_tests() # `TimmBackbone` has no `_init_weights`. Timm's way of weight init. seems to give larger magnitude in the intermediate values during `forward`. - def test_batching_equivalence(self, atol=1e-4, rtol=1e-4): + def test_batching_equivalence(self, atol=1e-3, rtol=1e-4): super().test_batching_equivalence(atol=atol, rtol=rtol) def test_timm_transformer_backbone_equivalence(self):