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

从tf2.0中删除了tf.contrib。从tf1.0迁移代码时，这可能是为数不多的需要更改的内容之一。但是，您可以在其他地方找到底层模块。示例here。

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

请仔细阅读how to migrate from tf1.0 to tf2.0上的官方文档。