小编rya*_*651的帖子

是否热修复Tensorflow模型无法通过.fit()在Eager模式下运行?

我正在尝试在Eager Execution中运行基本的CNN keras模型,但Tensorflow拒绝将模型视为急切的模型。我最初在稳定的1.13分支(最新)中尝试过此操作,请确保启用急切执行而没有结果。我升级到了2.0(最新),但是什么也没有。

模型

class CNN2(tf.keras.Model):

  def __init__(self, num_classes=7):
    super(CNN2, self).__init__()
    self.cnn1 = tf.keras.layers.Conv2D(32, (5,5), padding='same', strides=(2, 2),
                                      kernel_initializer='he_normal')
    self.bn1 = tf.keras.layers.BatchNormalization()
    self.cnn2 = tf.keras.layers.Conv2D(64, (5,5), padding='same', strides=(2, 2),
                                      kernel_initializer='he_normal')
    self.cnn3 = tf.keras.layers.Conv2D(128, (5,5), padding='same', strides=(2, 2),
                                      kernel_initializer='he_normal')
    self.bn2 = tf.keras.layers.BatchNormalization()
    self.pool = tf.keras.layers.MaxPooling2D((2,2))
    self.dnn1 = tf.keras.layers.Dense(128)
    self.dropout1 = tf.keras.layers.Dropout(0.45)
    self.flatten = tf.keras.layers.Flatten()
    self.dnn2 = tf.keras.layers.Dense(512)
    self.dnn3 = tf.keras.layers.Dense(256)
    self.classifier = tf.keras.layers.Dense(num_classes)    

  def simpleLoop(self, inputs, x):
        #x_Numpy = x.numpy(),
        for i, input in inputs:
            print("{0} - {1}".format(i,len(input)))             

  def call(self, inputs, …
Run Code Online (Sandbox Code Playgroud)

python machine-learning keras tensorflow eager-execution

5
推荐指数
2
解决办法
490
查看次数