update arguments to IR Block

pkgoogle · Jan 23, 2025 · 42e7901 · 42e7901
1 parent 8146370
commit 42e7901
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_hub/src/models/mobilenet/mobilenet_backbone.py b/keras_hub/src/models/mobilenet/mobilenet_backbone.py
@@ -684,7 +684,7 @@ def __init__(
                     ),
                     kernel_size=stackwise_kernel_size[block][inverted_block],
                     stride=stackwise_num_strides[block][inverted_block],
-                    se_ratio=stackwise_se_ratio[block][inverted_block],
+                    squeeze_excite_ratio=stackwise_se_ratio[block][inverted_block],
                     activation=stackwise_activation[block][inverted_block],
                     padding=stackwise_padding[block][inverted_block],
                     name=f"block_{block + 1}_{inverted_block}",