Skip to content

Commit b6502f4

Browse files
authored
Fix keras version problem (#15265)
* Fix keras version problem * Fix keras version problem * Fix keras version problem * Fix keras version problem * Fix keras version problem * Fix keras version problem * Fix keras version problem
1 parent a0e7d3e commit b6502f4

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

python/tvm/relay/frontend/keras.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1526,12 +1526,19 @@ def _convert_layer(keras_layer, etab, scope=""):
15261526
raise ValueError("Keras frontend currently supports tensorflow backend only.")
15271527
if keras.backend.image_data_format() != "channels_last":
15281528
raise ValueError("Keras frontend currently supports data_format = channels_last only.")
1529-
expected_model_class = keras.engine.training.Model
1530-
if hasattr(keras.engine, "InputLayer"):
1531-
input_layer_class = keras.engine.InputLayer
1529+
try:
1530+
import keras.engine as E
1531+
except ImportError:
1532+
try:
1533+
import keras.src.engine as E
1534+
except ImportError:
1535+
raise ImportError("Cannot find Keras's engine")
1536+
expected_model_class = E.training.Model
1537+
if hasattr(E, "InputLayer"):
1538+
input_layer_class = E.InputLayer
15321539
else:
15331540
# TFlite >=2.6
1534-
input_layer_class = keras.engine.input_layer.InputLayer
1541+
input_layer_class = E.input_layer.InputLayer
15351542
else:
15361543
# Importing from Tensorflow Keras (tf.keras)
15371544
try:

0 commit comments

Comments
 (0)