fix padding

Nguyendat-bit · Oct 30, 2021 · 8aa79b5 · 8aa79b5
1 parent f8dce5c
commit 8aa79b5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model.py b/model.py
@@ -15,21 +15,21 @@ def __Standard_Conv(self):
         return models.Sequential([
             Conv2D(filters= 32, kernel_size=(3,3), strides= (2,2), padding= 'valid'),
             BatchNormalization(),
-            Activation('relu')
+            ReLU(max_value= 6)
         ])
 
     def __Depthwise_Conv(self, strides, padding):
         return models.Sequential([
             DepthwiseConv2D(kernel_size= (3,3), strides= strides, padding= padding),
             BatchNormalization(),
-            Activation('relu')
+            ReLU(max_value= 6)
         ])
 
     def __Pointwise_Conv(self, filters):
         return models.Sequential([
             Conv2D(filters= int(filters * self._alpha), kernel_size= (1,1), strides= 1),
             BatchNormalization(),
-            Activation('relu')
+            ReLU(max_value= 6)
 
         ])