fix devastating TeacherUpdate mess up

(exclusive to branches develop2/3)
pyannote · Mar 18, 2022 · 66a873f · 66a873f
1 parent ddb7eb8
commit 66a873f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyannote/audio/tasks/segmentation/unsupervised_segmentation.py b/pyannote/audio/tasks/segmentation/unsupervised_segmentation.py
@@ -260,6 +260,7 @@ def __init__(
 
     def enqueue_teacher(self, teacher: OrderedDict[str, torch.Tensor]):
         if len(self.last_weights) >= self.average_of:
+            self.last_weights.pop(0)
             self.last_weights.append(teacher)
 
     def get_updated_weights(