因为tensorflow2.0版本与之前版本有所更新,故将代码修改即可:

#原 config = tf.ConfigProto(allow_soft_placement=True)
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
#原 sess = tf.Session(config=config)
sess =tf.compat.v1.Session(config=config)

或使用

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

替换

import tensorflow as tf