fix TeacherUpdate enqueue_teacher logic *again* ?

pyannote · Apr 4, 2022 · 94ea635 · 94ea635
1 parent 78ba279
commit 94ea635
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyannote/audio/tasks/segmentation/unsupervised_segmentation.py b/pyannote/audio/tasks/segmentation/unsupervised_segmentation.py
@@ -512,9 +512,9 @@ def __init__(
         self.teacher_weights_cache: OrderedDict[str, torch.Tensor] = None
 
     def enqueue_teacher(self, teacher: OrderedDict[str, torch.Tensor]):
+        self.last_weights.append(teacher)
         if len(self.last_weights) >= self.average_of:
             self.last_weights.pop(0)
-            self.last_weights.append(teacher)
 
     def get_updated_weights(
         self,