Update model_tensorflow.py

model_tensorflow.py

@@ -1,9 +1,7 @@
 import tensorflow as tf
 from tensorflow.keras import layers, Model
 
-# ==============================
-# Separable Convolution Block
-# ==============================
+
 def separable_block(x, filters, dropout_rate=0.25):
     x = layers.SeparableConv2D(filters, 3, padding="same", use_bias=False)(x)
     x = layers.BatchNormalization()(x)
@@ -19,24 +17,23 @@ def separable_block(x, filters, dropout_rate=0.25):
     return x
 
 
-# ==============================
+
 # Model Definition
-# ==============================
 def build_sara_tf_model(input_shape=(150, 150, 3), num_classes=6):
 
     inputs = tf.keras.Input(shape=input_shape)
 
-    # Stem
+   
     x = layers.Conv2D(32, 3, padding="same", use_bias=False)(inputs)
     x = layers.BatchNormalization()(x)
     x = layers.Activation("relu")(x)
 
-    # Feature extractor
+    
     x = separable_block(x, 64, 0.25)
     x = separable_block(x, 128, 0.25)
     x = separable_block(x, 256, 0.30)
 
-    # Head
+    
     x = layers.GlobalAveragePooling2D()(x)
 
     x = layers.Dense(128, use_bias=False)(x)