From 6d6ab318eafa68e283a80def868d25f37b29932e Mon Sep 17 00:00:00 2001 From: Ilya Trushkin Date: Tue, 28 Oct 2025 12:11:19 +0300 Subject: [PATCH] Extend supported pretrained checkpoints list for EfficientNet --- .../native/models/classification/backbones/efficientnet.py | 2 +- .../models/classification/backbones/test_otx_efficientnet.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/library/src/otx/backend/native/models/classification/backbones/efficientnet.py b/library/src/otx/backend/native/models/classification/backbones/efficientnet.py index 3754be782ff..76c30bbdc4d 100644 --- a/library/src/otx/backend/native/models/classification/backbones/efficientnet.py +++ b/library/src/otx/backend/native/models/classification/backbones/efficientnet.py @@ -672,6 +672,6 @@ def __new__( if pretrained: cache_dir = Path.home() / ".cache" / "torch" / "hub" / "checkpoints" - download_model(net=model, model_name=f"{model_name}", local_model_store_dir_path=str(cache_dir)) + download_model(net=model, model_name=f"{model_name}c", local_model_store_dir_path=str(cache_dir)) print(f"Download model weight in {cache_dir!s}") return model diff --git a/library/tests/unit/backend/native/models/classification/backbones/test_otx_efficientnet.py b/library/tests/unit/backend/native/models/classification/backbones/test_otx_efficientnet.py index efa89681dd4..ccc0b5f6b46 100644 --- a/library/tests/unit/backend/native/models/classification/backbones/test_otx_efficientnet.py +++ b/library/tests/unit/backend/native/models/classification/backbones/test_otx_efficientnet.py @@ -23,8 +23,9 @@ class TestOTXEfficientNet: "efficientnet_b8", ], ) - def test_forward(self, model_name): - model = EfficientNetBackbone(model_name, pretrained=None) + @pytest.mark.parametrize("pretrained", [True, False]) + def test_forward(self, model_name, pretrained): + model = EfficientNetBackbone(model_name, pretrained=pretrained) assert model(torch.randn(1, 3, 244, 244))[0].shape[-1] == 8 assert model(torch.randn(1, 3, 244, 244))[0].shape[-2] == 8