deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tolong

Browse Source
This commit is contained in:
James Betker 2022-05-23 08:27:54 -06:00
parent 9f16b25ce5
commit 40f844657b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/mel2vec.py
View File

@ -641,7 +641,7 @@ class ContrastiveTrainingWrapper(nn.Module):
mel[:, freq_start:freq_start+freq_mask_width] = 0
# Build input masks from inp_lengths if possible.
attention_mask = torch.ones_like(mel)
attention_mask = torch.ones_like(mel).long()
if inp_lengths is not None:
inp_lengths = inp_lengths // self.inp_length_factor
for i, l in enumerate(inp_lengths):