1. ホーム
  2. python

[解決済み】 tf.contrib.layers.xavier_initializer() を 2.0.0 に変更する。

2022-02-05 13:02:29

質問

を変更するにはどうすればよいですか?

tf.contrib.layers.xavier_initializer()

to tf version >= 2.0.0 ?

すべてのコード

W1 = tf.get_variable("W1", shape=[self.input_size, h_size],
                             initializer=tf.contrib.layers.xavier_initializer())

解決方法は?

のTF2代替品です。 tf.contrib.layers.xavier_initializer()tf.keras.initializers.glorot_normal (XavierとGlorotは同じイニシャライザーアルゴリズムの2つの名前です) ドキュメントリンク .

互換性のために dtype が重要な場合、以下のように使用します。 tf.compat.v1.keras.initializers.glorot_normal