fix devastating TeacherUpdate mess up

(exclusive to branches develop2/3)
pyannote · Mar 18, 2022 · 372f66c · 372f66c
1 parent 38e7dcb
commit 372f66c
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
@@ -218,6 +218,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(