tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update data_utils.py

Browse Source
This commit is contained in:
Stardust·减
2023-09-23 20:24:20 +08:00
committed by GitHub
parent fc3923def5
commit 732e9ce1e3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
data_utils.py
View File

@@ -208,7 +208,7 @@ class TextAudioSpeakerCollate:
torch.LongTensor([x[1].size(1) for x in batch]), dim=0, descending=True torch.LongTensor([x[1].size(1) for x in batch]), dim=0, descending=True
) )
max_text_len = max([len(x[0]) for x in batch]) max_text_len = max([batch[ids_sorted_decreasing[i]][7].size(1) for i in range(len(ids_sorted_decreasing))] + [len(x[0]) for x in batch])
max_spec_len = max([x[1].size(1) for x in batch]) max_spec_len = max([x[1].size(1) for x in batch])
max_wav_len = max([x[2].size(1) for x in batch]) max_wav_len = max([x[2].size(1) for x in batch])