FLOPS verbose=False

2020-04-26 16:31:57 -07:00 · 2020-04-26 16:31:57 -07:00 · 4a4bfb20de
parent a0a3bab9e6
commit 4a4bfb20de
1 changed files with 1 additions and 1 deletions
--- 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 = ''