Merge pull request #6 from phillies/fix/init_bits

fix bits parameter in init function

pytorch_modelsize.py

@@ -12,7 +12,7 @@ def __init__(self, model, input_size=(1,1,32,32), bits=32):
         '''
         self.model = model
         self.input_size = input_size
-        self.bits = 32
+        self.bits = bits
         return
 
     def get_parameter_sizes(self):