소스 검색

Update word2vec.py

bug fix for contrib.deprecated eliminatation in tf version 12.
Mostafa Rahmani 9 년 전
부모
커밋
edd4ba11d4
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      tutorials/embedding/word2vec.py

+ 1 - 1
tutorials/embedding/word2vec.py

@@ -365,7 +365,7 @@ class Word2Vec(object):
       self._word2id[w] = i
     true_logits, sampled_logits = self.forward(examples, labels)
     loss = self.nce_loss(true_logits, sampled_logits)
-    tf.contrib.deprecated.scalar_summary("NCE loss", loss)
+    tf.scalar_summary("NCE loss", loss)
     self._loss = loss
     self.optimize(loss)