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

fix(metric): add metric dtype control #533

Merged
merged 3 commits into from
Dec 11, 2023

Conversation

Pryest
Copy link
Contributor

@Pryest Pryest commented Dec 11, 2023

Motivation

avoid oom in metric calculation using fp32.

Modification

add a gpc config option to change dtype for metric calculation.

BC-break

now default using model type as dtype for metric cacluation

@gaoyang07 gaoyang07 self-requested a review December 11, 2023 08:20
@gaoyang07 gaoyang07 added the enhancement New feature or request label Dec 11, 2023
@Pryest Pryest force-pushed the fix/add_metric_dtype_control branch from 7c8e9b1 to fdce50a Compare December 11, 2023 09:43
@gaoyang07 gaoyang07 merged commit cc5b153 into InternLM:develop Dec 11, 2023
16 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants