code formatting

keras-team · Oct 24, 2024 · e4c43dc · e4c43dc
1 parent 0ea6a4c
commit e4c43dc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/keras_hub/src/utils/timm/convert_mobilenet.py b/keras_hub/src/utils/timm/convert_mobilenet.py
@@ -31,7 +31,7 @@ def convert_backbone_config(timm_config):
             [None, None],
             [0.25, 0.25, 0.25],
             [0.25, 0.25],
-            [0.25, 0.25, 0.25], 
+            [0.25, 0.25, 0.25],
         ]
         stackwise_activation = [
             ["relu", "relu"],
@@ -100,7 +100,7 @@ def port_batch_normalization(keras_layer_name, hf_weight_prefix):
             backbone.get_layer(keras_layer_name).moving_variance,
             hf_weight_key=f"{hf_weight_prefix}.running_var",
         )
-        
+
     # Stem
     port_conv2d("input_conv", "conv_stem")
     port_batch_normalization("input_batch_norm", "bn1")
@@ -154,6 +154,7 @@ def port_batch_normalization(keras_layer_name, hf_weight_prefix):
     # if version == "v2":
     # port_batch_normalization("output_batch_norm", "bn2")
 
+
 def convert_head(task, loader, timm_config):
     prefix = "classifier."
     loader.port_weight(