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

NU-Wave2 reproduction #4

Closed
yxlu-0102 opened this issue Nov 1, 2023 · 2 comments
Closed

NU-Wave2 reproduction #4

yxlu-0102 opened this issue Nov 1, 2023 · 2 comments

Comments

@yxlu-0102
Copy link

Hello, may I ask how many steps you have trained NU-Wave2 for? I couldn't achieve the metrics mentioned in the paper using their official checkpoint. If it's convenient for you, could you provide the checkpoint for your trained nuwave2?

@yoyolicoris
Copy link
Member

@yxlu-0102 Thanks for asking.
You can download the checkpoints from here:
https://drive.google.com/drive/folders/1guK563LP-h9NWd7ohLWP2_wAg_KUmeZD?usp=sharing
Unfortunately, I forgot which one I used in my paper 😅
I picked the one that perform the best.
It's probably the epoch=71 or epoch=72 one, but definitely not epoch=73.

I have to admit that their model performance fluctuates a lot during training. It's better to consistently monitor the process and pick the best one on the fly.

@yxlu-0102
Copy link
Author

It did work! Thanks a million! 😭

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

2 participants