tensorflow - Eager 模式优化器

Question

Eager 模式仅支持 TF native 优化器我在以下尝试过的每个优化器中都会遇到此错误:

def create_model():
    model = tf.keras.models.Sequential([tf.keras.layers.Dense(512,activation=tf.nn.relu, input_shape=(784,)), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10, activation=tf.nn.softmax) ])

    opt = tf.train.GradientDescentOptimizer
    model.compile(optimizer = opt, 
           loss=tf.keras.losses.sparse_categorical_crossentropy, metrics= ['accuracy']) 

    return model

所以我的问题是，什么是“TF native 优化器”？谢谢。

Answer 1

简短回答:从 opt = tf.train.GradientDescentOptimizer 更改至opt = tf.train.GradientDescentOptimizer(<your desired learning rate>) .

更长的答案:在上面提供的代码片段中，您将一个类( tf.train.GradientDescentOptimizer )而不是对象传递给 model.compile 。因此，错误消息提示 opt 的类型说法不正确。

希望有帮助。

(A recent commit 希望在未来版本中产生更好的错误消息)