diff --git a/coremltools/optimize/torch/pruning/magnitude_pruner.py b/coremltools/optimize/torch/pruning/magnitude_pruner.py index 6d68eb531..d837fc635 100644 --- a/coremltools/optimize/torch/pruning/magnitude_pruner.py +++ b/coremltools/optimize/torch/pruning/magnitude_pruner.py @@ -304,7 +304,7 @@ def __attrs_post_init__(self): if self.initial_sparsity is not None and self.initial_sparsity > 0.0: raise ValueError( f"Received initial_sparsity = {self.initial_sparsity} and " - f"n_m_ratio = {self.nm_ratio}. When n_m_ratio != None, the only allowed " + f"n_m_ratio = {self.n_m_ratio}. When n_m_ratio != None, the only allowed " f"value of initial_sparsity is 0." )