StyleTTS2 icon indicating copy to clipboard operation
StyleTTS2 copied to clipboard

Published 20 hours ago • verified publisher icon yl4579

HELP WANTED!!!!!!!!!!!

Open 21sK1p opened this issue 9 months ago • 3 comments

Can we add a checkpoint to the styleTTS2 model to finetune the model from the last epoch we trained? As I am using the free version of Colab, I can only finish my data to 5 epochs. After that, I exceeded my GPU limit, and all my data was saved in my drive, but then I had to start training it again from the start. I want to train it from my last checkpoint (last epoch trained). I asked if we could add a checkpoint to the model for the above query, just like tortoiseTTS (I asked blackbox.ai, and it says we can, but I need clarification on it)? cc @yl4579 @astricks @kmn1024 @eltociear @kmn1024

21sK1p avatar May 03 '24 10:05 21sK1p

Just edit the config to point the pretrained model to your last checkpoint, and set load_only_params to false.

godspirit00 avatar May 08 '24 11:05 godspirit00

Just edit the config to point the pretrained model to your last checkpoint, and set load_only_params to false.

it is not working. I tried what you said, but it is not working. @godspirit00

21sK1p avatar May 09 '24 11:05 21sK1p

Could you give us your config and the command you're running ?

Karesto avatar May 10 '24 15:05 Karesto

@21sK1p if you resolved this issue, could you please close it?

martinambrus avatar Sep 01 '24 18:09 martinambrus