fix train

2026-09-24 23:33:08 +08:00 · 2021-12-18 08:25:28 +00:00
parent dc7d4b00cd
commit 3312d6248e
1 changed files with 1 additions and 1 deletions
@@ -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)