From 4a4bfb20deba3c88ce12c44e7837eada95a42e25 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Apr 2020 16:31:57 -0700 Subject: [PATCH] FLOPS verbose=False --- 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 c6690924..215772bf 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -107,7 +107,7 @@ def model_info(model, verbose=False): try: # FLOPS from thop import profile - macs, _ = profile(model, inputs=(torch.zeros(1, 3, 480, 640),)) + macs, _ = profile(model, inputs=(torch.zeros(1, 3, 480, 640),), verbose=False) fs = ', %.1f GFLOPS' % (macs / 1E9 * 2) except: fs = ''