pythontensorflow

Tensorflow._api.v2.train has no attribute 'AdamOptimizer'


When using

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

in my Jupyter Notebook the following Error pops up:

module 'tensorflow._api.v2.train' has no attribute 'AdamOptimizer'

Tensorflow Version: 2.0.0-alpha0


Do you think the only possibility is to downgrade the TF version?


Solution

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

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