原因是TensorFlow 2.x之后把contrib这个库取消了,contrib的内容可以集成到下面三个包中

tf.keras.layers.Layer
tf.keras.Model
tf.Module

发现L2正则化存在于tf.keras里面,于是如下操作更改

我出错的代码是

regularizer = tf.contrib.layers.l2_regularizer(scale = 1e-10)
initializer = tf.contrib.layers.xavier_initializer()

更改为下面这两行就可以了

regularizer = tf.keras.regularizers.l2(1e-10)
initializer = tf.keras.initializers.glorot_normal()

代码就可以运行了

GitHub 加速计划 / te / tensorflow
184.54 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:23 天前 )
a49e66f2 PiperOrigin-RevId: 663726708 1 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 1 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐