diff --git a/4-Object_Detection/YOLOV3/core/common.py b/4-Object_Detection/YOLOV3/core/common.py index 88cc8884..fc473e1f 100644 --- a/4-Object_Detection/YOLOV3/core/common.py +++ b/4-Object_Detection/YOLOV3/core/common.py @@ -21,7 +21,7 @@ class BatchNormalization(tf.keras.layers.BatchNormalization): and `beta` will not be updated ! """ def call(self, x, training=False): - if not training: + if isinstance(training, bool) and training == False: training = tf.constant(False) training = tf.logical_and(training, self.trainable) return super().call(x, training)