mirror of https://github.com/coqui-ai/TTS.git
guided attn #9
This commit is contained in:
parent
80c18684c6
commit
8c4e03cebf
2
train.py
2
train.py
|
@ -106,7 +106,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
|
||||||
|
|
||||||
# create attention mask
|
# create attention mask
|
||||||
# TODO: vectorize
|
# TODO: vectorize
|
||||||
print(tet_input_var.shape)
|
print(text_input_var.shape)
|
||||||
print(mel_spec_var.shape)
|
print(mel_spec_var.shape)
|
||||||
N = text_input_var.shape[1]
|
N = text_input_var.shape[1]
|
||||||
T = mel_spec_var.shape[1]
|
T = mel_spec_var.shape[1]
|
||||||
|
|
Loading…
Reference in New Issue