Merge pull request #637 from shippingwang/fix-book-04

fix 04 hard code
PaddlePaddle · Oct 24, 2018 · 90dfa57 · 90dfa57
2 parents c666057 + b94b8b2
commit 90dfa57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/04.word2vec/train.py b/04.word2vec/train.py
@@ -38,7 +38,7 @@
 N = 5
 BATCH_SIZE = 100
 
-use_cuda = True  # set to True if training with GPU
+use_cuda = False  # set to True if training with GPU
 
 word_dict = paddle.dataset.imikolov.build_dict()
 dict_size = len(word_dict)