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

fix 04 hard code

Merge pull request #637 from shippingwang/fix-book-04
fix 04 hard code
90dfa576 · Tao Luo · GitHub · c6660577 · b94b8b2f · 90dfa576
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

04.word2vec/train.py 04.word2vec/train.py +1 -1

未找到文件。
--- a/04.word2vec/train.py
+++ b/04.word2vec/train.py
@@ -38,7 +38,7 @@ HIDDEN_SIZE = 256
 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)