提交 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,
chunk_scheme="IOB",
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):
test_target = chunk_evaluator.metrics + chunk_evaluator.states
inference_program = fluid.io.get_inference_program(test_target)
......
文件模式从 100644 更改为 100755
......@@ -9,7 +9,7 @@ from kpi import CostKpi, DurationKpi, AccKpi
train_acc_kpi = AccKpi('train_acc', 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 = [
train_acc_kpi,
......@@ -43,6 +43,6 @@ def log_to_ce(log):
if __name__ == '__main__':
log = sys.stdin.read()
print("*****")
print log
print(log)
print("****")
log_to_ce(log)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册