mirror of https://github.com/coqui-ai/TTS.git
Bug fix
This commit is contained in:
parent
b198541a65
commit
bf7590c7d2
|
@ -139,8 +139,7 @@ class MyDataset(Dataset):
|
||||||
mel_lengths = torch.LongTensor(mel_lengths)
|
mel_lengths = torch.LongTensor(mel_lengths)
|
||||||
stop_targets = torch.FloatTensor(stop_targets)
|
stop_targets = torch.FloatTensor(stop_targets)
|
||||||
|
|
||||||
return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs[
|
return text, text_lenghts, linear, mel, mel_lengths, stop_targets, item_idxs
|
||||||
0]
|
|
||||||
|
|
||||||
raise TypeError(("batch must contain tensors, numbers, dicts or lists;\
|
raise TypeError(("batch must contain tensors, numbers, dicts or lists;\
|
||||||
found {}".format(type(batch[0]))))
|
found {}".format(type(batch[0]))))
|
||||||
|
|
Loading…
Reference in New Issue