提交 cdcff5fd 编写于 作者: Y Yibing Liu

Set random_see before network creating in ce

上级 0c3e4835
...@@ -55,6 +55,9 @@ def main(train_data_file, ...@@ -55,6 +55,9 @@ def main(train_data_file,
word_dict_len = len(word_dict) word_dict_len = len(word_dict)
label_dict_len = len(label_dict) label_dict_len = len(label_dict)
if "CE_MODE_X" in os.environ:
fluid.default_startup_program().random_seed = 110
avg_cost, feature_out, word, mark, target = ner_net( avg_cost, feature_out, word, mark, target = ner_net(
word_dict_len, label_dict_len, parallel) word_dict_len, label_dict_len, parallel)
...@@ -98,8 +101,6 @@ def main(train_data_file, ...@@ -98,8 +101,6 @@ def main(train_data_file,
feeder = fluid.DataFeeder(feed_list=[word, mark, target], place=place) feeder = fluid.DataFeeder(feed_list=[word, mark, target], place=place)
exe = fluid.Executor(place) exe = fluid.Executor(place)
if "CE_MODE_X" in os.environ:
fluid.default_startup_program().random_seed = 110
exe.run(fluid.default_startup_program()) exe.run(fluid.default_startup_program())
embedding_name = 'emb' embedding_name = 'emb'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册