mirror of https://github.com/coqui-ai/TTS.git
guided attn #11
This commit is contained in:
parent
830a051a78
commit
e87d7d8d26
2
train.py
2
train.py
|
@ -107,7 +107,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
|
||||||
# create attention mask
|
# create attention mask
|
||||||
# TODO: vectorize
|
# TODO: vectorize
|
||||||
N = text_input_var.shape[1]
|
N = text_input_var.shape[1]
|
||||||
T = mel_spec_var.shape[1] / c.r
|
T = mel_spec_var.shape[1] // c.r
|
||||||
M = np.zeros([N, T])
|
M = np.zeros([N, T])
|
||||||
for t in range(T):
|
for t in range(T):
|
||||||
for n in range(N):
|
for n in range(N):
|
||||||
|
|
Loading…
Reference in New Issue