NaN during training when using own dataset

[cjay42]

While fine-tuning works as expected, doing regular training with a dataset that isn't LJSpeech would eventually cause a NaN loss at some point.
The culprit appears to be the following line, which causes a division by zero if wav happens to contain perfect silence:

hifigan/hifigan/dataset.py

Line 106 in 374a456

wav = flip * gain * wav / wav.abs().max()

I'm not sure what the best solution for this would be, as a quick fix I simply clipped the divisor so it can't reach zero:

wav = flip * gain * wav / max([wav.abs().max(), 0.001])

[joan126]

met same issue with you!!