fix train

3312d624 · LDOUBLEV · dc7d4b00 · 3312d624
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

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