fix train

3312d624 · LDOUBLEV · dc7d4b00 · 3312d624
Commit 3312d624 authored Dec 18, 2021 by LDOUBLEV
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- 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)