如何在 TensorFlow 上进行 Xavier 初始化

2023-12-23

我正在将我的 Caffe 网络移植到 TensorFlow,但它似乎没有 xavier 初始化。我在用着truncated_normal但这似乎让训练变得更加困难。


从 0.8 版本开始,有一个 Xavier 初始化器,请参阅此处的文档 https://www.tensorflow.org/versions/r0.11/api_docs/python/contrib.layers.html#xavier_initializer.

你可以使用这样的东西:

W = tf.get_variable("W", shape=[784, 256],
           initializer=tf.contrib.layers.xavier_initializer())
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何在 TensorFlow 上进行 Xavier 初始化 的相关文章

随机推荐