Merge pull request #212 from alibaba-damo-academy/dev_lhn

Update iterable_dataset.py
This commit is contained in:
hnluo 2023-03-10 20:53:33 +08:00 committed by GitHub
commit 716ea81703
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -247,7 +247,7 @@ class IterableESPnetDataset(IterableDataset):
if _type == "sound":
if self.mc:
data[name] = array.transpose(0, 1)
data[name] = array.transpose((1, 0))
else:
data[name] = array[0]
else:
@ -354,7 +354,7 @@ class IterableESPnetDataset(IterableDataset):
array = array.numpy()
if _type == "sound":
if self.mc:
data[name] = array.transpose(0, 1)
data[name] = array.transpose((1, 0))
else:
data[name] = array[0]
else: