提交 c0f8360e 编写于 作者: M mmglove 提交者: zhang wenhui

Add word2vec ce (#4163)

* add compress ce

* add ce for word2vec
上级 dd51d09a
......@@ -84,6 +84,9 @@ def parse_args():
required=False,
default=False,
help='print speed or not , (default: False)')
parser.add_argument(
'--enable_ce', action='store_true', help='If set, run the task with continuous evaluation logs.')
return parser.parse_args()
......@@ -195,6 +198,11 @@ def GetFileList(data_path):
def train(args):
# add ce
if args.enable_ce:
SEED = 102
fluid.default_main_program().random_seed = SEED
fluid.default_startup_program().random_seed = SEED
if not os.path.isdir(args.model_output_dir):
os.mkdir(args.model_output_dir)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册