From da96d14eaaecbb252b3dc922b1e27f86feac9e62 Mon Sep 17 00:00:00 2001
From: Tilman Kamp <5991088+tilmankamp@users.noreply.github.com>
Date: Fri, 19 Jun 2020 11:10:09 +0200
Subject: [PATCH] Fix #3089 - Recreate overlay queue on augmentation restart

---
 training/deepspeech_training/util/augmentations.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/training/deepspeech_training/util/augmentations.py b/training/deepspeech_training/util/augmentations.py
index a86b60a9..b7033c51 100644
--- a/training/deepspeech_training/util/augmentations.py
+++ b/training/deepspeech_training/util/augmentations.py
@@ -244,11 +244,12 @@ class Overlay(SampleAugmentation):
         self.source = source
         self.snr = float_range(snr)
         self.layers = int_range(layers)
-        self.queue = Queue(max(1, math.floor(self.probability * self.layers[1] * os.cpu_count())))
         self.current_sample = None
+        self.queue = None
         self.enqueue_process = None
 
     def start(self, buffering=BUFFER_SIZE):
+        self.queue = Queue(max(1, math.floor(self.probability * self.layers[1] * os.cpu_count())))
         self.enqueue_process = Process(target=_enqueue_overlay_samples,
                                        args=(self.source, self.queue),
                                        kwargs={'buffering': buffering})
@@ -285,6 +286,9 @@ class Overlay(SampleAugmentation):
     def stop(self):
         if self.enqueue_process is not None:
             self.enqueue_process.terminate()
+            self.enqueue_process = None
+        self.current_sample = None
+        self.queue = None
 
 
 class Codec(SampleAugmentation):