Skip to content

Add config option to specify float precision type during training#170

Open
OrianeN wants to merge 1 commit intofacebookresearch:mainfrom
OrianeN:custom_fprecision
Open

Add config option to specify float precision type during training#170
OrianeN wants to merge 1 commit intofacebookresearch:mainfrom
OrianeN:custom_fprecision

Conversation

@OrianeN
Copy link
Copy Markdown

@OrianeN OrianeN commented Nov 2, 2023

Not all CUDA devices support mixed-type precision - myself I got this error when launching train.py:

RuntimeError: Current CUDA Device does not support bfloat16. Please switch dtype to float16.

Therefore I propose this PR to let the user choose which type of precision they want to use during training.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Nov 2, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants