tensorflow|module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘【TensorFlow 2.x】解决方法
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 2.x的方法tf.initializers.GlorotUniform()进行初始化
文章图片
??注意:方法二中的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.
推荐阅读
- python|Python3入门机器学习经典算法与应用——knn算法数据归一化 Feature Scaling
- python|一文述说人工智能(AI)发展史,几经沉浮!
- 算法|计算机视觉近10年发展回顾与未来发展方向
- tensorflow|Anaconda安装tensorflow及tensorflow-gpu
- 论文分享|CVPR 2022 | 未知目标检测模块STUD(学习视频中的未知目标)
- 机器学习|Dynamic Graph Learning-Neural Network for Multivariate Time Series Modeling
- 时空数据预测(基于图神经网络)|图神经网络应用变体(时空数据挖掘一)
- 深度学习实战案例|【图像分类案例】(9) MobileNetV3 癌症图像二分类,附Pytorch完整代码
- #|李宏毅机器学习笔记4 -- DNN训练