From 0271f9a1b7c5224395140c53074c4b66be3b6854 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Thu, 31 Oct 2024 14:57:43 -0700 Subject: [PATCH 1/8] add closest variants --- .../src/utils/timm/convert_efficientnet.py | 16 ++++++++++++++ .../convert_efficientnet_checkpoints.py | 21 +++++++++++++++++++ 2 files changed, 37 insertions(+) diff --git a/keras_hub/src/utils/timm/convert_efficientnet.py b/keras_hub/src/utils/timm/convert_efficientnet.py index 609c26d355..5d179b3bbb 100644 --- a/keras_hub/src/utils/timm/convert_efficientnet.py +++ b/keras_hub/src/utils/timm/convert_efficientnet.py @@ -18,6 +18,22 @@ "width_coefficient": 1.0, "depth_coefficient": 1.1, }, + "b2": { + "width_coefficient": 1.1, + "depth_coefficient": 1.2, + }, + "b3": { + "width_coefficient": 1.2, + "depth_coefficient": 1.4, + }, + "b4": { + "width_coefficient": 1.4, + "depth_coefficient": 1.8, + }, + "b5": { + "width_coefficient": 1.6, + "depth_coefficient": 2.2, + }, } diff --git a/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py b/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py index 5790d6130c..0437103779 100644 --- a/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py +++ b/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py @@ -3,8 +3,22 @@ python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ --preset efficientnet_b0_ra_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b0_ra_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b0_ra4_e3600_r224_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ --preset efficientnet_b1_ft_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b1_ft_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b1_ra4_e3600_r240_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b2_ra_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b2_ra_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b3_ra2_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b3_ra2_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b4_ra2_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b4_ra2_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b5_sw_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b5_sw_imagenet +python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ + --preset efficientnet_b5_sw_ft_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b5_sw_ft_imagenet """ import os @@ -22,7 +36,14 @@ PRESET_MAP = { "efficientnet_b0_ra_imagenet": "timm/efficientnet_b0.ra_in1k", + "efficientnet_b0_ra4_e3600_r224_imagenet": "timm/efficientnet_b0.ra4_e3600_r224_in1k", "efficientnet_b1_ft_imagenet": "timm/efficientnet_b1.ft_in1k", + "efficientnet_b1_ra4_e3600_r240_imagenet": "timm/efficientnet_b1.ra4_e3600_r240_in1k", + "efficientnet_b2_ra_imagenet": "timm/efficientnet_b2.ra_in1k", + "efficientnet_b3_ra2_imagenet": "timm/efficientnet_b3.ra2_in1k", + "efficientnet_b4_ra2_imagenet": "timm/efficientnet_b4.ra2_in1k", + "efficientnet_b5_sw_imagenet": "timm/efficientnet_b5.sw_in12k", + "efficientnet_b5_sw_ft_imagenet": "timm/efficientnet_b5.sw_in12k_ft_in1k", } FLAGS = flags.FLAGS From cb752d0fd4a03653e3ff8f7e3ec0e5288f7abea1 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Mon, 4 Nov 2024 11:12:42 -0800 Subject: [PATCH 2/8] added to presets file --- .../efficientnet/efficientnet_presets.py | 100 +++++++++++++++++- 1 file changed, 99 insertions(+), 1 deletion(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 39c9514816..473e097e27 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -14,10 +14,25 @@ }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet", }, + "efficientnet_b0_ra4_e3600_r224_imagenet": { + "metadata": { + "description": ( + "EfficientNet B0 model pre-trained on the ImageNet 1k dataset by" + " Ross Wightman. Trained with timm scripts using hyper-parameters" + " inspired by the MobileNet-V4 small, mixed with go-to hparams " + "from timm and ""ResNet Strikes Back""." + ), + "params": 5288548, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet", + }, "efficientnet_b1_ft_imagenet": { "metadata": { "description": ( - "EfficientNet B1 model fine-trained on the ImageNet 1k dataset." + "EfficientNet B1 model pre-trained on the ImageNet 1k dataset." ), "params": 7794184, "official_name": "EfficientNet", @@ -26,4 +41,87 @@ }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet", }, + "efficientnet_b1_ra4_e3600_r240_imagenet": { + "metadata": { + "description": ( + "EfficientNet B1 model pre-trained on the ImageNet 1k dataset by" + " Ross Wightman. Trained with timm scripts using hyper-parameters" + " inspired by the MobileNet-V4 small, mixed with go-to hparams " + "from timm and ""ResNet Strikes Back""." + ), + "params": 7794184, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet", + }, + "efficientnet_b2_ra_imagenet": { + "metadata": { + "description": ( + "EfficientNet B2 model pre-trained on the ImageNet 1k dataset " + "with RandAugment recipe." + ), + "params": 9109994, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet", + }, + "efficientnet_b3_ra2_imagenet": { + "metadata": { + "description": ( + "EfficientNet B3 model pre-trained on the ImageNet 1k dataset " + "with RandAugment2 recipe." + ), + "params": 12233232, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet", + }, + "efficientnet_b4_ra2_imagenet": { + "metadata": { + "description": ( + "EfficientNet B4 model pre-trained on the ImageNet 1k dataset " + "with RandAugment2 recipe." + ), + "params": 19341616, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet", + }, + "efficientnet_b5_sw_imagenet": { + "metadata": { + "description": ( + "EfficientNet B5 model pre-trained on the ImageNet 12k dataset " + "by Ross Wightman. Based on Swin Transformer train / pretrain " + "recipe with modifications (related to both DeiT and ConvNeXt recipes)." + ), + "params": 30389784, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_imagenet", + }, + "efficientnet_b5_sw_ft_imagenet": { + "metadata": { + "description": ( + "EfficientNet B5 model pre-trained on the ImageNet 12k dataset " + "and fine-tuned on ImageNet-1k by Ross Wightman. Based on Swin " + "Transformer train / pretrain recipe with modifications " + "(related to both DeiT and ConvNeXt recipes)." + ), + "params": 30389784, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_ft_imagenet", + }, } From 63420d63beee25e4e22ea88a0084d018d4481fa5 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Mon, 4 Nov 2024 11:14:17 -0800 Subject: [PATCH 3/8] format pass --- keras_hub/src/models/efficientnet/efficientnet_presets.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 473e097e27..4e36931531 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -20,7 +20,9 @@ "EfficientNet B0 model pre-trained on the ImageNet 1k dataset by" " Ross Wightman. Trained with timm scripts using hyper-parameters" " inspired by the MobileNet-V4 small, mixed with go-to hparams " - "from timm and ""ResNet Strikes Back""." + "from timm and " + "ResNet Strikes Back" + "." ), "params": 5288548, "official_name": "EfficientNet", @@ -47,7 +49,9 @@ "EfficientNet B1 model pre-trained on the ImageNet 1k dataset by" " Ross Wightman. Trained with timm scripts using hyper-parameters" " inspired by the MobileNet-V4 small, mixed with go-to hparams " - "from timm and ""ResNet Strikes Back""." + "from timm and " + "ResNet Strikes Back" + "." ), "params": 7794184, "official_name": "EfficientNet", From 319bcf1aedaa734d33279a2215160d14b3b1ed25 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Mon, 4 Nov 2024 11:17:00 -0800 Subject: [PATCH 4/8] fix quote literal and formatting --- keras_hub/src/models/efficientnet/efficientnet_presets.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 4e36931531..5cb9e93826 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -20,9 +20,7 @@ "EfficientNet B0 model pre-trained on the ImageNet 1k dataset by" " Ross Wightman. Trained with timm scripts using hyper-parameters" " inspired by the MobileNet-V4 small, mixed with go-to hparams " - "from timm and " - "ResNet Strikes Back" - "." + 'from timm and "ResNet Strikes Back".' ), "params": 5288548, "official_name": "EfficientNet", @@ -49,9 +47,7 @@ "EfficientNet B1 model pre-trained on the ImageNet 1k dataset by" " Ross Wightman. Trained with timm scripts using hyper-parameters" " inspired by the MobileNet-V4 small, mixed with go-to hparams " - "from timm and " - "ResNet Strikes Back" - "." + 'from timm and "ResNet Strikes Back".' ), "params": 7794184, "official_name": "EfficientNet", From a844601cdab11afc85f7c8229df700c7c1ac4155 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Mon, 4 Nov 2024 11:32:31 -0800 Subject: [PATCH 5/8] fix uris --- .../convert_efficientnet_checkpoints.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py b/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py index 0437103779..c4192a63aa 100644 --- a/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py +++ b/tools/checkpoint_conversion/convert_efficientnet_checkpoints.py @@ -2,23 +2,23 @@ Convert efficientnet checkpoints. python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b0_ra_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b0_ra_imagenet + --preset efficientnet_b0_ra_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b0_ra4_e3600_r224_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet + --preset efficientnet_b0_ra4_e3600_r224_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b1_ft_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b1_ft_imagenet + --preset efficientnet_b1_ft_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b1_ra4_e3600_r240_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet + --preset efficientnet_b1_ra4_e3600_r240_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b2_ra_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b2_ra_imagenet + --preset efficientnet_b2_ra_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b3_ra2_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b3_ra2_imagenet + --preset efficientnet_b3_ra2_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b4_ra2_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b4_ra2_imagenet + --preset efficientnet_b4_ra2_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b5_sw_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b5_sw_imagenet + --preset efficientnet_b5_sw_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b5_sw_imagenet python tools/checkpoint_conversion/convert_efficientnet_checkpoints.py \ - --preset efficientnet_b5_sw_ft_imagenet --upload_uri kaggle://kerashub/efficientnet/keras/efficientnet_b5_sw_ft_imagenet + --preset efficientnet_b5_sw_ft_imagenet --upload_uri kaggle://keras/efficientnet/keras/efficientnet_b5_sw_ft_imagenet """ import os From 11023a80d2a5a5dcceb59abecba8d56b2fc8f430 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Thu, 14 Nov 2024 10:28:07 -0800 Subject: [PATCH 6/8] add version to kaggle handles --- .../efficientnet/efficientnet_presets.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 5cb9e93826..a85eeab0ce 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -12,7 +12,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra_imagenet/1", }, "efficientnet_b0_ra4_e3600_r224_imagenet": { "metadata": { @@ -27,7 +27,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b0_ra4_e3600_r224_imagenet/1", }, "efficientnet_b1_ft_imagenet": { "metadata": { @@ -39,7 +39,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ft_imagenet/1", }, "efficientnet_b1_ra4_e3600_r240_imagenet": { "metadata": { @@ -54,7 +54,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b1_ra4_e3600_r240_imagenet/1", }, "efficientnet_b2_ra_imagenet": { "metadata": { @@ -67,7 +67,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet/1", }, "efficientnet_b3_ra2_imagenet": { "metadata": { @@ -80,7 +80,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet/1", }, "efficientnet_b4_ra2_imagenet": { "metadata": { @@ -93,7 +93,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet/1", }, "efficientnet_b5_sw_imagenet": { "metadata": { @@ -107,7 +107,7 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_imagenet/1", }, "efficientnet_b5_sw_ft_imagenet": { "metadata": { @@ -122,6 +122,6 @@ "path": "efficientnet", "model_card": "https://arxiv.org/abs/1905.11946", }, - "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_ft_imagenet", + "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b5_sw_ft_imagenet/1", }, } From 020a2e54a09b81f2906ed6d5a227f2714e6d8557 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Thu, 14 Nov 2024 10:40:42 -0800 Subject: [PATCH 7/8] fix preset config --- .../src/models/efficientnet/efficientnet_presets.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index 2b88493fc2..c5df699a4d 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -63,6 +63,10 @@ "with RandAugment recipe." ), "params": 9109994, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b2_ra_imagenet/1", }, "efficientnet_b3_ra2_imagenet": { @@ -72,6 +76,10 @@ "with RandAugment2 recipe." ), "params": 12233232, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b3_ra2_imagenet/1", }, "efficientnet_b4_ra2_imagenet": { @@ -81,6 +89,10 @@ "with RandAugment2 recipe." ), "params": 19341616, + "official_name": "EfficientNet", + "path": "efficientnet", + "model_card": "https://arxiv.org/abs/1905.11946", + }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_b4_ra2_imagenet/1", }, "efficientnet_b5_sw_imagenet": { @@ -125,7 +137,6 @@ }, "kaggle_handle": "kaggle://keras/efficientnet/keras/efficientnet_el_ra_imagenet/1", }, - "efficientnet_em_ra2_imagenet": { "metadata": { "description": ( From 4bbbae4cc99025bd90532daaa61354c7891a4383 Mon Sep 17 00:00:00 2001 From: Piseth Ky Date: Thu, 14 Nov 2024 10:44:04 -0800 Subject: [PATCH 8/8] fix description error --- keras_hub/src/models/efficientnet/efficientnet_presets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_hub/src/models/efficientnet/efficientnet_presets.py b/keras_hub/src/models/efficientnet/efficientnet_presets.py index c5df699a4d..0807f7e163 100644 --- a/keras_hub/src/models/efficientnet/efficientnet_presets.py +++ b/keras_hub/src/models/efficientnet/efficientnet_presets.py @@ -32,7 +32,7 @@ "efficientnet_b1_ft_imagenet": { "metadata": { "description": ( - "EfficientNet B1 model pre-trained on the ImageNet 1k dataset." + "EfficientNet B1 model fine-tuned on the ImageNet 1k dataset." ), "params": 7794184, "official_name": "EfficientNet",