diff --git a/04.word2vec/train.py b/04.word2vec/train.py index 6385fbb9..229d0184 100644 --- 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)