OperatorNotAllowedInGraphError: использование tf.Tensor в качестве Python bool не допускается при выполнении Graph - PullRequest
1 голос
/ 18 января 2020

Эй, я пытаюсь преобразовать свою модель Keras в CoreML, но получаю эту ошибку:

tensorflow.python.framework.errors_impl.OperatorNotAllowedInGraphError: using a tf.Tensor as 
a Python bool is not allowed in Graph execution. Use Eager execution or decorate this 
function with @tf.function.

Вот мой код:

import keras
import coremltools
import tensorflow as tf

model = keras.models.load_model('machine.h5')

print(model.input)
print(model.input_shape)
print(model.output)

model= coremltools.converters.keras.convert(
'machine.h5', input_name_shape_dict={'embedding_input:0': (None, 24)},
output_names=['dense_2/Sigmoid:0'])

model.save('xcodeModel.mlmodel')

Не уверен, что я здесь неправильно. Надеюсь, на это есть ответ!

1 Ответ

0 голосов
/ 23 февраля 2020

Я думаю, что ошибка из-за Tensorflow и используемой вами версии Keras. Пожалуйста, понизьте тензор потока до версии 1.14.0, а Keras - до версии 2.2.4, возможно, это решит проблему.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...