From 108334db29e45a80f43b12a8dde24712257290cc Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 30 Mar 2020 16:04:08 -0700 Subject: [PATCH] FLOPs update --- utils/torch_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/torch_utils.py b/utils/torch_utils.py index c194e0f0..e4490a70 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -89,9 +89,9 @@ def model_info(model, verbose=False): (i, name, p.requires_grad, p.numel(), list(p.shape), p.mean(), p.std())) print('Model Summary: %g layers, %g parameters, %g gradients' % (len(list(model.parameters())), n_p, n_g)) - # FLOPS report + # FLOPs # from thop import profile - # macs, params = profile(model, inputs=(torch.zeros(1, 3, 608, 608),)) + # macs, params = profile(model, inputs=(torch.zeros(1, 3, 640, 640),)) # print('%.3f GFLOPs' % (macs / 1E9 * 2))