Skip to content

Commit

Permalink
Improve error message, issue #284
Browse files Browse the repository at this point in the history
  • Loading branch information
Dobiasd committed Jul 5, 2021
1 parent 48912c3 commit 6f9e4b1
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion keras_export/convert_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -598,7 +598,8 @@ def get_all_weights(model, prefix):
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."
assert node.call_kwargs["training"] is not True, \
"training=true is not supported, see https://github.com/Dobiasd/frugally-deep/issues/284"
layer_type = type(layer).__name__
name = prefix + layer.name
assert is_ascii(name)
Expand Down

0 comments on commit 6f9e4b1

Please sign in to comment.