updates

2018-12-11 21:49:56 +01:00
parent 7fb729269b
commit c591936446
3 changed files with 5 additions and 7 deletions
@@ -33,7 +33,7 @@ def model_info(model):  # Plots a line-by-line description of a PyTorch model
    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('%5g %50s %9s %12g %20s %12g %12g' % (
+        print('%5g %50s %9s %12g %20s %12.3g %12.3g' % (
            i, name, p.requires_grad, p.numel(), list(p.shape), p.mean(), p.std()))
    print('Model Summary: %g layers, %g parameters, %g gradients\n' % (i + 1, n_p, n_g))