refine

de6bc445 · MissPenguin · 89496d4a · de6bc445
Commit de6bc445 authored Jun 21, 2021 by MissPenguin
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/program.py tools/program.py +1 -1

No files found.
--- a/tools/program.py
+++ b/tools/program.py
@@ -276,7 +276,7 @@ def train(config,
                    valid_dataloader,
                    post_process_class,
                    eval_class,
-                    "table",
+                    model_type,
                    use_srn=use_srn)
                cur_metric_str = 'cur metric, {}'.format(', '.join(
                    ['{}: {}'.format(k, v) for k, v in cur_metric.items()]))