fix nm_ratio (#2023)

ZhouXiaolin · Zelaren · web-flow · commit dd907474e1ba · 2023-10-23T15:59:23.000-07:00
Co-authored-by: zhouxiaolin &lt;zhouxiaolinux@gmail.com&gt;
diff --git 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."
                 )
 

Original file line number	Diff line number	Diff line change
`@@ -304,7 +304,7 @@ def __attrs_post_init__(self):`
`304`	`304`	`if self.initial_sparsity is not None and self.initial_sparsity > 0.0:`
`305`	`305`	`raise ValueError(`
`306`	`306`	`f"Received initial_sparsity = {self.initial_sparsity} and "`
`307`		`- f"n_m_ratio = {self.nm_ratio}. When n_m_ratio != None, the only allowed "`
	`307`	`+ f"n_m_ratio = {self.n_m_ratio}. When n_m_ratio != None, the only allowed "`
`308`	`308`	`f"value of initial_sparsity is 0."`
`309`	`309`	`)`
`310`	`310`