update

2026-09-09 11:13:26 +08:00 · 2019-10-22 14:58:44 -06:00
parent 2f40f04c62
commit d0042d4d47
1 changed files with 1 additions and 1 deletions
@@ -85,7 +85,7 @@ with tf.device('/gpu:0'):
        def __init__(self, vocab_size=vocab_size, embedding_size=embedding_dim, **kwargs):
            super(TiedEmbeddingSoftmax, self).__init__()
            self.w = self.add_weight(name='w', shape=(vocab_size, embedding_size),
-                                     initializer='random_normal',
+                                     initializer='random_uniform',
                                     trainable=True)
            self.b = self.add_weight(name='b', shape=(vocab_size,),
                                     initializer='zeros',