fix code spell in diff_mutable_module

open-mmlab · Aug 3, 2022 · 1eeb7fa · 1eeb7fa
1 parent fecdcc6
commit 1eeb7fa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmrazor/models/mutables/mutable_module/diff_mutable_module.py b/mmrazor/models/mutables/mutable_module/diff_mutable_module.py
@@ -290,7 +290,7 @@ class DiffChoiceRoute(DiffMutableModule[str, List[str]]):
     def __init__(
         self,
         edges: nn.ModuleDict,
-        num_chsoen: int = 2,
+        num_chosen: int = 2,
         with_arch_param: bool = False,
         alias: Optional[str] = None,
         init_cfg: Optional[Dict] = None,
@@ -303,7 +303,7 @@ def __init__(
         self._with_arch_param = with_arch_param
         self._is_fixed = False
         self._candidates: nn.ModuleDict = edges
-        self.num_chosen = num_chsoen
+        self.num_chosen = num_chosen
 
     def forward_fixed(self, inputs: Union[List, Tuple]) -> Tensor:
         """Forward when the mutable is in `fixed` mode.