From 6e825acb722f8d9df05a40a0486286dacee016d3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 23 Nov 2018 15:32:41 +0100 Subject: [PATCH] updates --- utils/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index b472d4ca..990cc115 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -22,12 +22,12 @@ def load_classes(path): def model_info(model): # Plots a line-by-line description of a PyTorch model n_p = sum(x.numel() for x in model.parameters()) # number parameters n_g = sum(x.numel() for x in model.parameters() if x.requires_grad) # number gradients - print('\n%4s %70s %9s %12s %20s %12s %12s' % ('', 'name', 'gradient', 'parameters', 'shape', 'mu', 'sigma')) + print('\n%5s %50s %9s %12s %20s %12s %12s' % ('layer', 'name', 'gradient', 'parameters', 'shape', 'mu', 'sigma')) for i, (name, p) in enumerate(model.named_parameters()): name = name.replace('module_list.', '') - print('%4g %70s %9s %12g %20s %12g %12g' % ( + print('%5g %50s %9s %12g %20s %12g %12g' % ( i, name, p.requires_grad, p.numel(), list(p.shape), p.mean(), p.std())) - print('\nModel Summary: %g layers, %g parameters, %g gradients\n' % (i + 1, n_p, n_g)) + print('Model Summary: %g layers, %g parameters, %g gradients\n' % (i + 1, n_p, n_g)) def class_weights(): # frequency of each class in coco train2014