TensorFlow 2.x - tf.contrib.layers.variance_scaling_initializer()

Question

我对这条线和使用 Tensorflow 2.3.1 有疑问

if w_init is None:
        w_init = tf.contrib.layers.variance_scaling_initializer()

AttributeError: 模块 'tensorflow' 没有属性 'contrib'

Answer 1

tf.contrib 已从 tf2.0 中删除。当从 tf1.0 迁移代码时，它可能是少数必须更改的事情之一。但是，您可以在其他地方找到底层模块。示例here。

if w_init is None:
        w_init = tf.keras.initializers.variance_scaling()

请查看how to migrate from tf1.0 to tf2.0 上的官方文档。