未验证 提交 2a0c3d4d 编写于 作者: X xiaoting 提交者: GitHub

fix eval mode without srn (#1889)

* fix base model

* fix start time
上级 e40100c5
......@@ -326,9 +326,12 @@ def eval(model, valid_dataloader, post_process_class, eval_class):
if idx >= len(valid_dataloader):
break
images = batch[0]
others = batch[-4:]
start = time.time()
preds = model(images, others)
if "SRN" in str(model.head):
others = batch[-4:]
preds = model(images, others)
else:
preds = model(images)
batch = [item.numpy() for item in batch]
# Obtain usable results from post-processing methods
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册