search

NVIDIA / flowtron

Flowtron is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer
https://nv-adlr.github.io/Flowtron
Apache License 2.0
889 stars 177 forks source link

Saving state_dict in the checkpoint? #18

Closed adrianastan closed 4 years ago

adrianastan commented 4 years ago

Hi,

I was wondering why aren't you saving the state_dict in the checkpoints? It doesn't seem like you are saving it somewhere else. And this is also the case for Waveglow: https://github.com/NVIDIA/waveglow/blob/master/train.py

Thanks!

rafaelvalle commented 4 years ago

When running multiple experiments we find it useful to save the model in case there are changes to the code.

adrianastan commented 4 years ago

Ok, got it, thanks!