updates

utils/torch_utils.py

@@ -25,6 +25,7 @@ def select_device(force_cpu=False):
                 print("           device%g _CudaDeviceProperties(name='%s', total_memory=%dMB)" %
                       (i, x[i].name, x[i].total_memory / c))
 
+    print('')  # skip a line
     return device
 
 

utils/utils.py

@@ -41,7 +41,7 @@ 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%5s %40s %9s %12s %20s %10s %10s' % ('layer', 'name', 'gradient', 'parameters', 'shape', 'mu', 'sigma'))
+    print('%5s %40s %9s %12s %20s %10s %10s' % ('layer', 'name', 'gradient', 'parameters', 'shape', 'mu', 'sigma'))
     for i, (name, p) in enumerate(model.named_parameters()):
         name = name.replace('module_list.', '')
         print('%5g %40s %9s %12g %20s %10.3g %10.3g' % (