Skip to content

GitLab

Commit edd4ba11 authored by Mostafa Rahmani's avatar Mostafa Rahmani Committed by GitHub
Browse files

Update word2vec.py 

bug fix for contrib.deprecated eliminatation in tf version 12.
parent 75d3de64
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tutorials/embedding/word2vec.py
View file @ edd4ba11
...@@ -365,7 +365,7 @@ class Word2Vec(object): ...@@ -365,7 +365,7 @@ class Word2Vec(object):
self._word2id[w] = i self._word2id[w] = i
true_logits, sampled_logits = self.forward(examples, labels) true_logits, sampled_logits = self.forward(examples, labels)
loss = self.nce_loss(true_logits, sampled_logits) 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._loss = loss
self.optimize(loss) self.optimize(loss)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment