diff --git a/generate_sequence_by_rnn_lm/config.py b/generate_sequence_by_rnn_lm/config.py index 2085b7ba5f7448fbad26794e30e004b42543ca9a..487885a95d77878fcb347205372a2f8c636dd4e6 100644 --- a/generate_sequence_by_rnn_lm/config.py +++ b/generate_sequence_by_rnn_lm/config.py @@ -26,7 +26,7 @@ num_passes = 20 # how many passes to train the model log_period = 50 save_period_by_batches = 50 -use_gpu = True # to use gpu or not +use_gpu = False # to use gpu or not trainer_count = 1 # number of trainer ################## for model configuration ################## diff --git a/generate_sequence_by_rnn_lm/train.py b/generate_sequence_by_rnn_lm/train.py index 2958592748c0fe982972717017dede11c04ebb7e..90125d278fee34f73935e6d8cf940729a8f6a0c6 100644 --- a/generate_sequence_by_rnn_lm/train.py +++ b/generate_sequence_by_rnn_lm/train.py @@ -35,8 +35,7 @@ def train(topology, os.mkdir(model_save_dir) # initialize PaddlePaddle - paddle.init( - use_gpu=conf.use_gpu, gpu_id=3, trainer_count=conf.trainer_count) + paddle.init(use_gpu=conf.use_gpu, trainer_count=conf.trainer_count) # create optimizer adam_optimizer = paddle.optimizer.Adam( diff --git a/nmt_without_attention/train.py b/nmt_without_attention/train.py index 9600df8e5b70cca90543062b040e6dddc540440c..5604d70d9f796d2561977e166931d092902e25b2 100644 --- a/nmt_without_attention/train.py +++ b/nmt_without_attention/train.py @@ -63,4 +63,4 @@ def train(save_dir_path, source_dict_dim, target_dict_dim): if __name__ == '__main__': - train(save_dir_path="models", source_dict_dim=3000, target_dict_dim=3000) + train(save_dir_path="models", source_dict_dim=30000, target_dict_dim=30000)