From 199483df41a70ce89bbb63961949a3fc6809c81e Mon Sep 17 00:00:00 2001 From: zhouxiaolin Date: Mon, 23 Oct 2023 10:43:23 +0800 Subject: [PATCH] fix nm_ratio --- coremltools/optimize/torch/pruning/magnitude_pruner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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." )