Procházet zdrojové kódy

Merge pull request #891 from DorianKodelja/patch-1

Updating Softmax bias comment
Neal Wu před 8 roky
rodič
revize
e6d4e082e9
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  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 [].