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

training issue:loss=nan when use 2B model #82

Open

ucaswindlike opened this issue Nov 27, 2024 · 2 comments

ucaswindlike commented Nov 27, 2024

when using 2b model,loss=nan after first step. using 5B model did not meet this issue.

Collaborator

bubbliiiing commented Nov 28, 2024

We haven't encountered this situation yet. Are you using BF16?

Author

ucaswindlike commented Nov 28, 2024

yes,encounter this when using bf16, while fp32 is running correctly

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment