Fix additional 0-dim accesses

distiller/pruning/sensitivity_pruner.py

@@ -45,7 +45,7 @@ class SensitivityPruner(_ParameterPruner):
 
     def set_param_mask(self, param, param_name, zeros_mask_dict, meta):
         if not hasattr(param, 'stddev'):
-            param.stddev = torch.std(param).data[0]
+            param.stddev = torch.std(param).item()
 
         if param_name not in self.sensitivities:
             if '*' not in self.sensitivities:

distiller/quantization/q_utils.py

@@ -55,7 +55,7 @@ def linear_dequantize(input, scale_factor, inplace=False):
 
 
 def get_tensor_max_abs(tensor):
-    return max(abs(tensor.max().data[0]), abs(tensor.min().data[0]))
+    return max(abs(tensor.max().item()), abs(tensor.min().item()))
 
 
 def get_quantized_range(num_bits, signed=True):