Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

FLOPS verbose=False

This commit is contained in:
Glenn Jocher 2020-04-26 16:31:57 -07:00
parent a0a3bab9e6
commit 4a4bfb20de
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/torch_utils.py
View File

@ -107,7 +107,7 @@ def model_info(model, verbose=False):
try: # FLOPS try: # FLOPS
from thop import profile 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) fs = ', %.1f GFLOPS' % (macs / 1E9 * 2)
except: except:
fs = '' fs = ''