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.

Sign up for GitHub

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

Jump to bottom

我训练完成过后，我应该怎么使用？ #20

Open
wangaocheng opened this issue Nov 13, 2024 · 4 comments
Open

我训练完成过后，我应该怎么使用？ #20

wangaocheng opened this issue Nov 13, 2024 · 4 comments
Labels
good first issue Good for newcomers

Comments

@wangaocheng
Copy link

按照README的方法，我完成了stage1和stage2的训练，然后我应该如何使用我自己训练后的模型？
@DBDXSS
Copy link
Collaborator

DBDXSS commented Nov 14, 2024

您好，可以修改configs/inference/inference.yaml中的audio_ckpt_dir为stage2中您想使用的权重的path

@wangaocheng
Copy link
Author

@DBDXSS 感谢你的回复 只需要修改这一个参数就可以了 对吗？其他不用改了 对吗？stage1训练的模型 就没用了是吗？

@DBDXSS
Copy link
Collaborator

DBDXSS commented Nov 19, 2024

是的，1阶段的模型在2阶段有保留

@DBDXSS
Copy link
Collaborator

DBDXSS commented Nov 19, 2024

您可以直接输出net.pth中的权重keys查看是否包含全部需要的权重

@DBDXSS DBDXSS added the good first issue Good for newcomers label Nov 20, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
good first issue Good for newcomers
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wangaocheng @DBDXSS