refine

de6bc445 · MissPenguin · 89496d4a · de6bc445
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- 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()]))