[Fix] Fix typo in BaseDistiller (open-mmlab#73)

pppppM · Feb 11, 2022 · b18976f · b18976f
1 parent b28136b
commit b18976f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmrazor/models/distillers/base.py b/mmrazor/models/distillers/base.py
@@ -120,7 +120,7 @@ def __init__(self, align_methods=None, **kwargs):
             self.context_manager = ConversionContext(align_methods)
 
     @abstractmethod
-    def prepare_from_student(self, supernet):
+    def prepare_from_student(self, student):
         """Register forward hooks to students and teachers."""
         pass