search

sh-lee-prml / HierSpeechpp

The official implementation of HierSpeech++
MIT License
1.13k stars 134 forks source link

Missing keys in checkpoint file #37

Closed arsm closed 4 months ago

arsm commented 4 months ago

Thank you for your excellent work! I'm currently working on fine-tuning the TTV model with a non-English dataset. I've downloaded the 'ttv_lt960_ckpt.pth' and attempted to load it using utils.load_checkpoint function in the 'train_ttv_v1.py' to initiate the model. However, the checkpoint doesn't include the required keys like 'optimizer', 'model', 'learning_rate', and 'iteration', making it impossible to load. Could you please guide me on what might be missing or if there are any additional steps I need to take?

sh-lee-prml commented 4 months ago

Hi!

Here is a checkpoint including all.

https://drive.google.com/file/d/1ZCgvuBFAmY39r8dENWdOglXyp6iVmati/view?usp=drive_link

Thanks!

arsm commented 4 months ago

Thank you so much, @sh-lee-prml. I appreciate it.