From d0042d4d47b67d7994db18a43309f2ed346fe29c Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 14:58:44 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 5b48610..e2f8f0b 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -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',