fix bug when rounding weights (lava-nc#683)

monkin77 · May 8, 2023 · 0650613 · 0650613
1 parent 552387a
commit 0650613
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/lava/utils/weightutils.py b/src/lava/utils/weightutils.py
@@ -239,6 +239,7 @@ def truncate_weights(weights: ty.Union[np.ndarray, spmatrix],
         weights = (weights >> num_truncate_bits) << num_truncate_bits
     elif isinstance(weights, spmatrix):
         weights.data = (weights.data >> num_truncate_bits) << num_truncate_bits
+        weights.eliminate_zeros()
 
     if sign_mode == SignMode.INHIBITORY:
         weights = -weights