From de6bc4458b664b6338efd8a2f16947c8261616c6 Mon Sep 17 00:00:00 2001
From: MissPenguin <lichenxia1991@163.com>
Date: Mon, 21 Jun 2021 12:33:19 +0000
Subject: [PATCH] refine

---
 tools/program.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/program.py b/tools/program.py
index d68e123a..0cc60c38 100755
--- 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()]))
-- 
GitLab