From 26ebe500365eafd5c7fdd9bade0ef95b6cc83d88 Mon Sep 17 00:00:00 2001
From: xiaowo1996 <429740343@qq.com>
Date: Wed, 24 Feb 2021 01:38:56 +0800
Subject: [PATCH] FLOPS min stride 32 (#2276)

Signed-off-by: xiaowo1996 <429740343@qq.com>
---
 utils/torch_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/torch_utils.py b/utils/torch_utils.py
index 2cb09e71ce71..1b1cc2038c55 100644
--- a/utils/torch_utils.py
+++ b/utils/torch_utils.py
@@ -205,7 +205,7 @@ def model_info(model, verbose=False, img_size=640):
 
     try:  # FLOPS
         from thop import profile
-        stride = int(model.stride.max()) if hasattr(model, 'stride') else 32
+        stride = max(int(model.stride.max()), 32) if hasattr(model, 'stride') else 32
         img = torch.zeros((1, model.yaml.get('ch', 3), stride, stride), device=next(model.parameters()).device)  # input
         flops = profile(deepcopy(model), inputs=(img,), verbose=False)[0] / 1E9 * 2  # stride GFLOPS
         img_size = img_size if isinstance(img_size, list) else [img_size, img_size]  # expand if int/float