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

Tiny fixes in text_cls ce

上级 b199f192
...@@ -53,7 +53,7 @@ def main(train_data_file, test_data_file, vocab_file, target_file, emb_file, ...@@ -53,7 +53,7 @@ def main(train_data_file, test_data_file, vocab_file, target_file, emb_file,
chunk_scheme="IOB", chunk_scheme="IOB",
num_chunk_types=int(math.ceil((label_dict_len - 1) / 2.0))) num_chunk_types=int(math.ceil((label_dict_len - 1) / 2.0)))
inference_program = fluid.default_main_program().clone() inference_program = fluid.default_main_program().clone(for_test=True)
with fluid.program_guard(inference_program): with fluid.program_guard(inference_program):
test_target = chunk_evaluator.metrics + chunk_evaluator.states test_target = chunk_evaluator.metrics + chunk_evaluator.states
inference_program = fluid.io.get_inference_program(test_target) inference_program = fluid.io.get_inference_program(test_target)
......
文件模式从 100644 更改为 100755
...@@ -9,7 +9,7 @@ from kpi import CostKpi, DurationKpi, AccKpi ...@@ -9,7 +9,7 @@ from kpi import CostKpi, DurationKpi, AccKpi
train_acc_kpi = AccKpi('train_acc', 0.005, actived=True) train_acc_kpi = AccKpi('train_acc', 0.005, actived=True)
train_cost_kpi = CostKpi('train_cost', 0.005, actived=True) train_cost_kpi = CostKpi('train_cost', 0.005, actived=True)
train_duration_kpi = DurationKpi('train_duration', 0.06, actived=True) train_duration_kpi = DurationKpi('train_duration', 0.05, actived=True)
tracking_kpis = [ tracking_kpis = [
train_acc_kpi, train_acc_kpi,
...@@ -43,6 +43,6 @@ def log_to_ce(log): ...@@ -43,6 +43,6 @@ def log_to_ce(log):
if __name__ == '__main__': if __name__ == '__main__':
log = sys.stdin.read() log = sys.stdin.read()
print("*****") print("*****")
print log print(log)
print("****") print("****")
log_to_ce(log) log_to_ce(log)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册