VoiceCraft icon indicating copy to clipboard operation
VoiceCraft copied to clipboard

training loss is nan

Open WoBuChiTang opened this issue 1 year ago • 2 comments

Hello, I am training Chinese data with a weight of 830m. At the beginning, the loss is normal, but after 2000 steps, there is a situation of "loss is nan" in a large area. May I ask if you have encountered it when training gigaspeech?

WoBuChiTang avatar Apr 09 '24 02:04 WoBuChiTang

use AdamW with smaller learning rate

see finetuning section in readme

jasonppy avatar Apr 09 '24 03:04 jasonppy

use AdamW with smaller learning rate

see finetuning section in readme

Thank you!

WoBuChiTang avatar Apr 09 '24 03:04 WoBuChiTang

Hi @WoBuChiTang! Have you fixed that error? Did suggestions (reducing the learning rate and using AdamW) from Jason help? If yes, can you please share hyperparameters that you have used?

Magauiya avatar Aug 28 '24 14:08 Magauiya