Tensorflow._api.v2.train没有属性“ AdamOptimizer”

时间:2019-03-23 20:54:18

标签: python tensorflow

使用

model.compile(optimizer = tf.train.AdamOptimizer(),
              loss = 'sparse_categorical_crossentropy',
              metrics=['accuracy'])

在我的Jupyter笔记本中弹出以下错误:

模块'tensorflow._api.v2.train'没有属性'AdamOptimizer'

Tensorflow版本: 2.0.0-alpha0


您认为唯一的可能是降级TF版本吗?

7 个答案:

答案 0 :(得分:14)

model.compile(optimizer = tf.keras.optimizers.Adam(),
              loss = 'sparse_categorical_crossentropy',
              metrics=['accuracy'])

答案 1 :(得分:4)

tf.optimizers.Adam()

是要走的路。没有理由降级。
与1.14相比,tf 2.0有很多更改。
请注意,Adam的参数名称也已更改。例如beta1现在是beta_1,请查看Meixu Songs链接中的文档。

答案 2 :(得分:2)

我有同样的错误。我删除了

tf.train.AdamOptimizer() 

我写了

tf.optimizers.Adam()

相反。

答案 3 :(得分:1)

tf.train.AdamOptimizer() => tf.optimizers.Adam()

来自https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/optimizers

答案 4 :(得分:1)

应该是:

tf.compat.v1.train.AdamOptimizer()

答案 5 :(得分:0)

我还没有尝试过2.0,但是从我在开发人员提交视频中看到的内容来看,您可以使用

model.compile(optimizer = 'adam',
           loss = 'sparse_categorical_crossentropy',
           metrics=['accuracy'])

答案 6 :(得分:0)

这是升级版本中的次要更改。

请使用:

model.compile(optimizer=tf.optimizers.Adam(), loss="sparse_categorical_crossentropy")

谢谢!