fix

Author: jacobkimmel

pytorch_modelsize.py

@@ -18,10 +18,11 @@ def __init__(self, model, input_size=(1,1,32,32), bits=32):
  def get_parameter_sizes(self):
  '''Get sizes of all parameters in `model`'''
  mods = list(self.model.modules())
+ sizes = []
+ 
  for i in range(1,len(mods)):
  m = mods[i]
  p = list(m.parameters())
- sizes = []
  for j in range(len(p)):
  sizes.append(np.array(p[j].size()))