diff options
Diffstat (limited to 'tensorflow/examples/udacity/5_word2vec.ipynb')
| -rw-r--r-- | tensorflow/examples/udacity/5_word2vec.ipynb | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/examples/udacity/5_word2vec.ipynb b/tensorflow/examples/udacity/5_word2vec.ipynb index 5bc8152b1c..c266488bde 100644 --- a/tensorflow/examples/udacity/5_word2vec.ipynb +++ b/tensorflow/examples/udacity/5_word2vec.ipynb @@ -421,7 +421,7 @@ "\n", "graph = tf.Graph()\n", "\n", - "with graph.as_default():\n", + "with graph.as_default(), tf.device('/cpu:0'):\n", "\n", " # Input data.\n", " train_dataset = tf.placeholder(tf.int32, shape=[batch_size])\n", |