提交 96531a75 编写于 作者: C caoying03

fix parameter settings that may cause bugs for users.

上级 0a27ca9d
...@@ -26,7 +26,7 @@ num_passes = 20 # how many passes to train the model ...@@ -26,7 +26,7 @@ num_passes = 20 # how many passes to train the model
log_period = 50 log_period = 50
save_period_by_batches = 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 trainer_count = 1 # number of trainer
################## for model configuration ################## ################## for model configuration ##################
......
...@@ -35,8 +35,7 @@ def train(topology, ...@@ -35,8 +35,7 @@ def train(topology,
os.mkdir(model_save_dir) os.mkdir(model_save_dir)
# initialize PaddlePaddle # initialize PaddlePaddle
paddle.init( paddle.init(use_gpu=conf.use_gpu, trainer_count=conf.trainer_count)
use_gpu=conf.use_gpu, gpu_id=3, trainer_count=conf.trainer_count)
# create optimizer # create optimizer
adam_optimizer = paddle.optimizer.Adam( adam_optimizer = paddle.optimizer.Adam(
......
...@@ -63,4 +63,4 @@ def train(save_dir_path, source_dict_dim, target_dict_dim): ...@@ -63,4 +63,4 @@ def train(save_dir_path, source_dict_dim, target_dict_dim):
if __name__ == '__main__': 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)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册