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

GFLOPs correction

This commit is contained in:
Glenn Jocher 2020-03-30 15:30:53 -07:00
parent de52a008a5
commit 105882b3c6
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

@ -92,7 +92,7 @@ def model_info(model, verbose=False):
# FLOPS report # FLOPS report
# from thop import profile # from thop import profile
# macs, params = profile(model, inputs=(torch.zeros(1, 3, 608, 608),)) # macs, params = profile(model, inputs=(torch.zeros(1, 3, 608, 608),))
# print('%.3f FLOPS' % (macs / 1E9 * 2)) # print('%.3f GFLOPs' % (macs / 1E9 * 2))
def load_classifier(name='resnet101', n=2): def load_classifier(name='resnet101', n=2):