问题描述:

使用tf 1.14.0 如下模块报错:

train_loss = tf.keras.metrics.Mean('train_loss', dtype=tf.float32)

AttributeError: module 'tensorflow._api.v1.keras.metrics' has no attribute 'Mean'

问题解决:

更新tf即可:

pip install -U tensorflow