提交 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
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 ##################
......
......@@ -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(
......
......@@ -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)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册