detach output from teacher

open-mmlab · Mar 2, 2022 · c19a8ec · c19a8ec
1 parent 0dd407a
commit c19a8ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmrazor/models/distillers/self_distiller.py b/mmrazor/models/distillers/self_distiller.py
@@ -68,7 +68,7 @@ def teacher_forward_output_hook(self, module, inputs, outputs):
             outputs (tuple): out of module
         """
         if self.training and getattr(self, 'is_teacher', None):
-            self.teacher_outputs[self.module2name[module]].append(outputs)
+            self.teacher_outputs[self.module2name[module]].append(outputs.detach())
 
     def student_forward_output_hook(self, module, inputs, outputs):
         """Save the output.