Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to save and load checkpoints #38

Closed
TsukimiRini opened this issue Jun 2, 2022 · 2 comments
Closed

how to save and load checkpoints #38

TsukimiRini opened this issue Jun 2, 2022 · 2 comments

Comments

@TsukimiRini
Copy link

Hi,
I'm running experiments with default settings using this command:

python -m train wandb=null experiment=sashimi-sc09

I'd like to get the checkpoint of the best model in the training process and then load it to generate .wav files. But i find the only checkpoint generated is a last.ckpt under the directory outputs/yyyy-mm-dd/xx-xx-xx/checkpoints, and i have no idea how to load it (since the checkpoint loading in sashimi/generation.py loads a .pt file)

@krandiash
Copy link
Contributor

Hi, thanks for bringing this issue to our attention. We will have an updated version of the generation script in the new v3 release (coming soon) that allows you to load the .ckpt files directly and generate from them.

@albertfgu
Copy link
Contributor

By the way, the updated generation script is available that allows generating from any new experiments run in this repository.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants