mirror of
https://github.com/wassname/Clover-Edition.git
synced 2026-10-30 12:50:18 +08:00
update
This commit is contained in:
1 parent
2f40f04c62
commit
d0042d4d47
1 file changed
+1
-1
@@ -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',
|
||||
|
||||
Reference in new issue
Block a user