提交 1444dbcb 编写于 作者: L lispczz

fix a deep speech 2 speed bug

上级 88edc4c9
...@@ -327,7 +327,7 @@ class DataGenerator(object): ...@@ -327,7 +327,7 @@ class DataGenerator(object):
shift_len = self._rng.randint(0, batch_size - 1) shift_len = self._rng.randint(0, batch_size - 1)
batch_manifest = zip(*[iter(manifest[shift_len:])] * batch_size) batch_manifest = zip(*[iter(manifest[shift_len:])] * batch_size)
self._rng.shuffle(batch_manifest) 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: if not clipped:
res_len = len(manifest) - shift_len - len(batch_manifest) res_len = len(manifest) - shift_len - len(batch_manifest)
batch_manifest.extend(manifest[-res_len:]) batch_manifest.extend(manifest[-res_len:])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册