Add test cases with training=True for BatchNormalization and Dropout

Dobiasd · Jul 6, 2021 · ff2a937 · ff2a937
1 parent 357d663
commit ff2a937
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras_export/generate_test_models.py b/keras_export/generate_test_models.py
@@ -180,6 +180,7 @@ def get_test_model_exhaustive():
     outputs.append(BatchNormalization()(Permute((3, 1, 2))(inputs[32])))
 
     outputs.append(BatchNormalization()(inputs[0]))
+    outputs.append(BatchNormalization()(inputs[0], training=True))
     outputs.append(BatchNormalization(axis=1)(inputs[0]))
     outputs.append(BatchNormalization(axis=2)(inputs[0]))
     outputs.append(BatchNormalization(axis=3)(inputs[0]))
@@ -223,6 +224,7 @@ def get_test_model_exhaustive():
     outputs.append(BatchNormalization(axis=5)(inputs[20]))
 
     outputs.append(Dropout(0.5)(inputs[4]))
+    outputs.append(Dropout(0.5)(inputs[4], training=True))
 
     outputs.append(ZeroPadding2D(2)(inputs[4]))
     outputs.append(ZeroPadding2D((2, 3))(inputs[4]))