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