Fix trainer checkpointing

lucidrains / naturalspeech2-pytorch

Implementation of Natural Speech 2, Zero-shot Speech and Singing Synthesizer, in Pytorch

MIT License

1.26k stars 100 forks source link

Closed janjurca closed 1 year ago

janjurca commented 1 year ago

Hi I ve made a little change to trainer in order to properly save checkpoints

lucidrains commented 1 year ago

@janjurca oh yes indeed it was borked! should be fixed

thank you!

janjurca commented 1 year ago

Great! Thanks for your work