From 3312d6248e81896fa6798fc17375b419e34b591c Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Sat, 18 Dec 2021 08:25:28 +0000 Subject: [PATCH] fix train --- tools/program.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/program.py b/tools/program.py index 538d0f62..84dc6c1d 100755 --- 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) -- GitLab