Merge pull request WZMIAOMIAO#181 from WZMIAOMIAO/dev

add training para
sue-up · Mar 15, 2021 · 193fe0e · 193fe0e
2 parents 45fad17 + 98f3f35
commit 193fe0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow_classification/Test6_mobilenet/model_v2.py b/tensorflow_classification/Test6_mobilenet/model_v2.py
@@ -56,11 +56,11 @@ def __init__(self, in_channel, out_channel, stride, expand_ratio, **kwargs):
         ])
         self.main_branch = Sequential(layer_list, name='expanded_conv')
 
-    def call(self, inputs, training=False):
+    def call(self, inputs, training=False, **kwargs):
         if self.use_shortcut:
             return inputs + self.main_branch(inputs, training=training)
         else:
-            return self.main_branch(inputs)
+            return self.main_branch(inputs, training=training)
 
 
 def MobileNetV2(im_height=224,