From 83c9cfb7de4c0f1359198da0de74daf9b9ac125b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 19 Mar 2020 12:30:37 -0700 Subject: [PATCH] FocalLoss() and obj loss speed and stability update --- 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 18664b3a..0b7013b0 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -88,7 +88,7 @@ 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)) - # Report FLOPS + # FLOPS report # from thop import profile # macs, params = profile(model, inputs=(torch.zeros(1, 3, 608, 608),)) # print('%.3f FLOPS' % (macs / 1E9 * 2))