fix train

2021-12-18 08:25:28 +00:00 · 2021-12-18 08:25:28 +00:00 · 3312d6248e
parent dc7d4b00cd
commit 3312d6248e
1 changed files with 1 additions and 1 deletions
--- a/tools/program.py
+++ b/tools/program.py
@ -239,7 +239,7 @@ def train(config,
            else:
                if model_type == 'table' or extra_input:
                    preds = model(images, data=batch[1:])
-                if model_type == "kie":
+                elif model_type == "kie":
                    preds = model(batch)
                else:
                    preds = model(images)