Open mingri-lt opened 1 year ago
I have fix this problem and use biaobei tts dataset to train success.
I have made some changes to the code in order to smooth out the training.
in validation.py :
def validate(hp, args, generator, discriminator, valloader, writer, step):
... ...
loader = tqdm.tqdm(valloader.dataset, desc='Validation loop')
... ...
for mel, audio in loader:
mel = mel.reshape(1, mel.size(0), mel.size(1))
audio = audio.reshape(1, 1, audio.size(1))
... ...
in train.py :
... ...
loader = tqdm.tqdm(trainloader.dataset, desc='Loading train data')
for (melG, audioG), (melD, audioD) in loader:
... ...
melG = melG.reshape(1, melG.size(0), melG.size(1))
audioG = audioG.reshape(1, audioG.size(0), audioG.size(1))
melD = melD.reshape(1, melD.size(0), melD.size(1))
audioD = audioD.reshape(1, audioD.size(0), audioD.size(1))
... ...
When I tried to train my own model, I ran into this error
Then I made a few changes: loader = tqdm.tqdm(valloader, desc='Validation loop') loader = tqdm.tqdm(valloader.dataset, desc='Validation loop')
and then:
How do I fix it please?