mirror of
https://github.com/coqui-ai/TTS.git
synced 2025-12-25 12:49:29 +01:00
bug fix
This commit is contained in:
2
train.py
2
train.py
@@ -111,7 +111,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
|
||||
# create attention mask
|
||||
N = text_input_var.shape[1]
|
||||
T = mel_spec_var.shape[1] // c.r
|
||||
M = create_attn_mask(N, T, g)
|
||||
M = create_attn_mask(N, T, 0.03)
|
||||
|
||||
# forward pass
|
||||
mel_output, linear_output, alignments =\
|
||||
|
||||
Reference in New Issue
Block a user