mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
Merge pull request #212 from alibaba-damo-academy/dev_lhn
Update iterable_dataset.py
This commit is contained in:
commit
716ea81703
@ -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:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user