tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法


module 'tensorflow.compat.v1' has no attribute 'contrib'【TensorFlow 2.x】解决方法

tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法
文章图片
tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法
文章图片

??这是我们报错的代码。原因是TensorFlow 2.x之后把contrib这个库取消了。xavier_initializer函数返回一个用于初始化权重的初始化程序 “Xavier” 。这个初始化器是用来保持每一层的梯度大小都差不多相同。


??下面有两种方法解决这种问题:
方法一:
??TensorFlow 2.x之后把tf.contrib.layers.xavier_initializer()替换成了 tf.keras.initializers.glorot_normal()(Xavier 和 Glorot 是对同一种初始化算法的不同命名),使用新的函数替换即可

tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法
文章图片



方法二:
??使用TensorFlow 2.x的方法tf.initializers.GlorotUniform()进行初始化

tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法
文章图片

??注意:方法二中的tf是(import tensorflow as tf)而不是(import tensorflow.compat.v1 as tf)





下面是我总结两种方法参考链接:
链接1: https://stackoverflow.com/questions/64255154/change-tf-contrib-layers-xavier-initializer-to-2-0-0.
【tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法】链接2: https://www.cnblogs.com/hale547/p/13574401.html.

    推荐阅读