From 105882b3c672729218c2b7cf5b9637aee98e284c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 30 Mar 2020 15:30:53 -0700 Subject: [PATCH] GFLOPs correction --- 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 56e5bba6..c194e0f0 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -92,7 +92,7 @@ def model_info(model, verbose=False): # FLOPS report # from thop import profile # 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):