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 = ''