Skip to content

GitLab

Unverified Commit 2a0c3d4d authored by xiaoting's avatar xiaoting Committed by GitHub
Browse files

fix eval mode without srn (#1889) 

* fix base model

* fix start time
parent e40100c5
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
tools/program.py
View file @ 2a0c3d4d
...@@ -326,9 +326,12 @@ def eval(model, valid_dataloader, post_process_class, eval_class): ...@@ -326,9 +326,12 @@ def eval(model, valid_dataloader, post_process_class, eval_class):
if idx >= len(valid_dataloader): if idx >= len(valid_dataloader):
break break
images = batch[0] images = batch[0]
others = batch[-4:]
start = time.time() 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] batch = [item.numpy() for item in batch]
# Obtain usable results from post-processing methods # Obtain usable results from post-processing methods
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment