experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #3090 from tilmankamp/augext

Browse Source 
Fix #3089 - Recreate overlay queue on augmentation restart
This commit is contained in:
Tilman Kamp 2020-06-19 12:22:34 +02:00 committed by GitHub
commit 6f2ba4b5b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
training/deepspeech_training/util/augmentations.py
View File

@ -244,11 +244,12 @@ class Overlay(SampleAugmentation):
self.source = source self.source = source
self.snr = float_range(snr) self.snr = float_range(snr)
self.layers = int_range(layers) 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.current_sample = None
self.queue = None
self.enqueue_process = None self.enqueue_process = None
def start(self, buffering=BUFFER_SIZE): 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, self.enqueue_process = Process(target=_enqueue_overlay_samples,
args=(self.source, self.queue), args=(self.source, self.queue),
kwargs={'buffering': buffering}) kwargs={'buffering': buffering})
@ -285,6 +286,9 @@ class Overlay(SampleAugmentation):
def stop(self): def stop(self):
if self.enqueue_process is not None: if self.enqueue_process is not None:
self.enqueue_process.terminate() self.enqueue_process.terminate()
self.enqueue_process = None
self.current_sample = None
self.queue = None
class Codec(SampleAugmentation): class Codec(SampleAugmentation):