提交 5e609c1e 编写于 作者: T tink2123

fix srn

上级 5e7ad64c
...@@ -403,7 +403,7 @@ def eval(model, ...@@ -403,7 +403,7 @@ def eval(model,
start = time.time() start = time.time()
if model_type == 'table' or extra_input: if model_type == 'table' or extra_input:
preds = model(images, data=batch[1:]) preds = model(images, data=batch[1:])
if model_type == "kie": elif model_type == "kie":
preds = model(batch) preds = model(batch)
else: else:
preds = model(images) preds = model(images)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册