Use FastSpeech
Model Has Been Defined
Number of TTS Parameters: 25367169
Load data to buffer
100% 1132/1132 [32:11<00:00, 1.71s/it]
cost 1931.72s to load all data into buffer.
Defined Optimizer and Loss Function.
---Start New Training---
Traceback (most recent call last):
File "train.py", line 193, in <module>
main(args)
File "train.py", line 106, in main
length_target=duration)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/parallel/data_parallel.py", line 143, in forward
return self.module(*inputs, **kwargs)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/content/drive/My Drive/fastspeech3/FastSpeech/model.py", line 42, in forward
mel_max_length=mel_max_length)
File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/content/drive/My Drive/fastspeech3/FastSpeech/modules.py", line 82, in forward
output = self.LR(x, target, mel_max_length=mel_max_length)
File "/content/drive/My Drive/fastspeech3/FastSpeech/modules.py", line 72, in LR
output = alignment @ x
RuntimeError: Expected tensor to have size 152 at dimension 1, but got size 91 for argument #2 'batch2' (while checking arguments for bmm)
hi @xcmyz
after successful run of preprocess.py
when i run train.py it gives following error