Merge branch 'dygraph' into dyg_db

2026-09-24 23:33:08 +08:00 · 2022-07-01 11:28:23 +08:00
parent 872d73daf5 9ae52fb3ce
commit 00e67e0bf2
79 changed files with 5796 additions and 1007 deletions
@@ -255,6 +255,8 @@ def train(config,
                with paddle.amp.auto_cast():
                    if model_type == 'table' or extra_input:
                        preds = model(images, data=batch[1:])
+                    elif model_type in ["kie", 'vqa']:
+                        preds = model(batch)
                    else:
                        preds = model(images)
            else: