[Improvement]

Init teacher weight by default
open-mmlab · Feb 7, 2023 · 9d65b9a · 9d65b9a
1 parent a27952d
commit 9d65b9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmrazor/models/algorithms/distill/configurable/single_teacher_distill.py b/mmrazor/models/algorithms/distill/configurable/single_teacher_distill.py
@@ -56,9 +56,9 @@ def __init__(self,
                             f'{type(teacher)}')
 
         self.teacher = teacher
+        self.teacher.init_weights()
         if teacher_ckpt:
             # avoid loaded parameters be overwritten
-            self.teacher.init_weights()
             _ = load_checkpoint(self.teacher, teacher_ckpt)
         self.teacher_trainable = teacher_trainable
         if not self.teacher_trainable: