Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

[dataset] add singal channel conf & processor #2439

Merged
merged 1 commit into from
Mar 24, 2024

Conversation

kakashidan
Copy link
Contributor

当shard/raw list中存在多通道数据,dataset会在padding部分报错,错误信息如下(or 类似)

    padded_wavs = pad_sequence(sorted_wavs, batch_first=True, padding_value=0)
  File "/opt/conda/lib/python3.10/site-packages/torch/nn/utils/rnn.py", line 399, in pad_sequence
    return torch._C._nn.pad_sequence(sequences, batch_first, padding_value)
RuntimeError: The size of tensor a (2) must match the size of tensor b (26645) at non-singleton dimension 1

This pr tries to fix the above problem

@Mddct Mddct merged commit 5006fb2 into wenet-e2e:main Mar 24, 2024
6 checks passed
@kakashidan kakashidan deleted the fix-single-channel branch March 25, 2024 09:11
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants