提交 3319d9b0 编写于 作者: F frankwhzhang

fix README, change train.py input format 2.0

上级 83424e13
...@@ -103,8 +103,8 @@ python train.py small_train.txt small_test.txt --use_cuda 0 --parallel 0 ...@@ -103,8 +103,8 @@ python train.py small_train.txt small_test.txt --use_cuda 0 --parallel 0
base_lr=0.01, # base learning rate base_lr=0.01, # base learning rate
batch_size=batch_size, batch_size=batch_size,
pass_num=10, # the number of passed for training pass_num=10, # the number of passed for training
use_cuda=True, # whether to use GPU card use_cuda=use_cuda, # whether to use GPU card
parallel=False, # whether to be parallel parallel=parallel, # whether to be parallel
model_dir="model_recall20", # directory to save model model_dir="model_recall20", # directory to save model
init_low_bound=-0.1, # uniform parameter initialization lower bound init_low_bound=-0.1, # uniform parameter initialization lower bound
init_high_bound=0.1) # uniform parameter initialization upper bound init_high_bound=0.1) # uniform parameter initialization upper bound
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册