From b1850c7dcafc3cdb4fdf002a902d55ea10db481e Mon Sep 17 00:00:00 2001
From: Maximilian Gangloff <71378343+m-gangloff@users.noreply.github.com>
Date: Mon, 12 Sep 2022 16:45:16 +0200
Subject: [PATCH] Fix updated image size not used (#739)

Co-authored-by: magangloff <71378343+magangloff@users.noreply.github.com>
---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 60154a69..894fa08e 100644
--- a/test.py
+++ b/test.py
@@ -59,7 +59,7 @@ def test(data,
         imgsz = check_img_size(imgsz, s=gs)  # check img_size
         
         if trace:
-            model = TracedModel(model, device, opt.img_size)
+            model = TracedModel(model, device, imgsz)
 
     # Half
     half = device.type != 'cpu' and half_precision  # half precision only supported on CUDA