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 遷移代碼時，它可能是少數必須更改的事情之一。 但是，您可以在其他地方找到底層模塊。 這里的例子。

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

通過這個關於如何從 tf1.0 遷移到 tf2.0 的官方文檔做 go 。