Disallow conversion of models having layers with training==True, issue …

…#284
Dobiasd · Jul 5, 2021 · 48912c3 · 48912c3
1 parent 144c757
commit 48912c3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/keras_export/convert_model.py b/keras_export/convert_model.py
@@ -596,6 +596,9 @@ def get_all_weights(model, prefix):
     layers = model.layers
     assert K.image_data_format() == 'channels_last'
     for layer in layers:
+        for node in layer.inbound_nodes:
+            if "training" in node.call_kwargs:
+                assert node.call_kwargs["training"] is not True, "Only inference mode is supported."
         layer_type = type(layer).__name__
         name = prefix + layer.name
         assert is_ascii(name)