refine

2021-06-21 12:33:19 +00:00 · 2021-06-21 12:33:19 +00:00 · de6bc4458b
parent 89496d4ab7
commit de6bc4458b
1 changed files with 1 additions and 1 deletions
--- 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()]))