From 960c80576db94b8edb89b0a10ddd6df45a6a44cf Mon Sep 17 00:00:00 2001 From: ydshieh Date: Thu, 30 Jan 2025 15:26:26 +0100 Subject: [PATCH] fix --- tests/models/timm_backbone/test_modeling_timm_backbone.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/models/timm_backbone/test_modeling_timm_backbone.py b/tests/models/timm_backbone/test_modeling_timm_backbone.py index 9701e9d68..a0ce2f59f 100644 --- a/tests/models/timm_backbone/test_modeling_timm_backbone.py +++ b/tests/models/timm_backbone/test_modeling_timm_backbone.py @@ -115,9 +115,9 @@ class TimmBackboneModelTest(ModelTesterMixin, BackboneTesterMixin, PipelineTeste def test_config(self): self.config_tester.run_common_tests() - @is_flaky( - description="`TimmBackbone` has no `_init_weights`. Timm's way of weight init. seems to give larger magnitude in the intermediate values during `forward`." - ) + # @is_flaky( + # description="`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): super().test_batching_equivalence(atol=atol, rtol=rtol)