Mask creator int -> round (#589)

* Fix: round for pruning index to match _threshold_from_sparsity

Author: KSGulin

src/sparseml/pytorch/optim/mask_creator_pruning.py

@@ -176,7 +176,7 @@ def create_sparsity_masks(
             local_rng = random.Random(42)
             local_rng.shuffle(rand_indices)
             num_elem = tensor.numel()
-            num_mask = int(num_elem * sparsity_target)
+            num_mask = round(num_elem * sparsity_target)
             rand_indices = rand_indices[:num_mask]
             rand_indices = tensor.new_tensor(rand_indices, dtype=torch.int64)
             zero_indices = zero_indices[rand_indices, :]

src/sparseml/pytorch/sparsification/pruning/mask_creator.py

@@ -140,7 +140,7 @@ def create_sparsity_masks(
             local_rng = random.Random(42)
             local_rng.shuffle(rand_indices)
             num_elem = tensor.numel()
-            num_mask = int(num_elem * sparsity_target)
+            num_mask = round(num_elem * sparsity_target)
             rand_indices = rand_indices[:num_mask]
             rand_indices = tensor.new_tensor(rand_indices, dtype=torch.int64)
             zero_indices = zero_indices[rand_indices, :]