refine

97a66874 · MissPenguin · 053cc43d · 97a66874
Commit 97a66874 authored Jun 22, 2021 by MissPenguin
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 4 deletions

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

No files found.
--- 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,
+def eval(model,
-         model_type, use_srn=False):
+         valid_dataloader,
+         post_process_class,
+         eval_class,
+         model_type,
+         use_srn=False):
    model.eval()
    with paddle.no_grad():
        total_frame = 0.0