refine

97a66874 · MissPenguin · 053cc43d · 97a66874
显示空白变更内容
内联并排

Showing with 8 addition and 4 deletion

tools/program.py tools/program.py +8 -4

未找到文件。
--- a/tools/program.py
+++ b/tools/program.py
@@ -338,8 +338,12 @@ def train(config,
    return


-def eval(model, valid_dataloader, post_process_class, eval_class,
-         model_type, use_srn=False):
+def eval(model,
+         valid_dataloader,
+         post_process_class,
+         eval_class,
+         model_type,
+         use_srn=False):
    model.eval()
    with paddle.no_grad():
        total_frame = 0.0