Browse Source

Merge pull request #891 from DorianKodelja/patch-1

Updating Softmax bias comment
Neal Wu 8 years ago
parent
commit
e6d4e082e9
1 changed files with 1 additions and 1 deletions
  1. 1 1
      tutorials/embedding/word2vec.py

+ 1 - 1
tutorials/embedding/word2vec.py

@@ -210,7 +210,7 @@ class Word2Vec(object):
         tf.zeros([opts.vocab_size, opts.emb_dim]),
         name="sm_w_t")
 
-    # Softmax bias: [emb_dim].
+    # Softmax bias: [vocab_size].
     sm_b = tf.Variable(tf.zeros([opts.vocab_size]), name="sm_b")
 
     # Global step: scalar, i.e., shape [].