при использовании keras сообщать об ошибке: используйте нетерпеливое выполнение или украсьте эту функцию @ tf.function - PullRequest
Новая
       68

при использовании keras сообщать об ошибке: используйте нетерпеливое выполнение или украсьте эту функцию @ tf.function

0 голосов
/ 13 июля 2020

Я не думаю, что в настоящее время tenorflow поддерживает создание tf.keras.Model, содержащего tf.cond или tf.while_loop.

Я построил модель на основе tf2.0, которая содержит while l oop, но я хочу сохранить его как файл pb, поэтому мне нужно использовать tf.keras.Model, чтобы перестроить мою модель.

Я пробовал код ниже, и он сообщил об ошибке: 

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 tensorflow as tf

layers = tf.keras.layers
a = layers.Input(shape=tf.TensorShape([]), dtype=tf.bool)
b = a[0]
c = tf.cond(b, lambda: 1, lambda: 0)
print (c)

Я использовал версию tenorflow v2.0.0-69-g765ac8d 2.0.1

В чем проблема с приведенным выше кодом?

...