From 1444dbcb0fb6f437a2bfb98def252da925580ce8 Mon Sep 17 00:00:00 2001 From: lispczz Date: Thu, 21 Sep 2017 16:02:43 +0800 Subject: [PATCH] fix a deep speech 2 speed bug --- deep_speech_2/data_utils/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_speech_2/data_utils/data.py b/deep_speech_2/data_utils/data.py index 8bff6826..7ddf1f33 100644 --- a/deep_speech_2/data_utils/data.py +++ b/deep_speech_2/data_utils/data.py @@ -327,7 +327,7 @@ class DataGenerator(object): shift_len = self._rng.randint(0, batch_size - 1) batch_manifest = zip(*[iter(manifest[shift_len:])] * batch_size) self._rng.shuffle(batch_manifest) - batch_manifest = list(sum(batch_manifest, ())) + batch_manifest = [item for batch in batch_manifest for item in batch] if not clipped: res_len = len(manifest) - shift_len - len(batch_manifest) batch_manifest.extend(manifest[-res_len:]) -- GitLab