From 2a0c3d4dac67cfd49e432303443bb9a50e75071f Mon Sep 17 00:00:00 2001
From: xiaoting <31891223+tink2123@users.noreply.github.com>
Date: Sun, 31 Jan 2021 22:37:30 +0800
Subject: [PATCH] fix eval mode without srn (#1889)

* fix base model

* fix start time
---
 tools/program.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tools/program.py b/tools/program.py
index 694d6415..f3ba4945 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -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