Unverified Commit 500fce07 authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[Trainer] add disclaimer that full_determinism is slow (#22368)

parent a0cbbba3
...@@ -538,7 +538,7 @@ class TrainingArguments: ...@@ -538,7 +538,7 @@ class TrainingArguments:
CUDA Out-of-Memory errors. Requires accelerate to be installed (`pip install accelerate`) CUDA Out-of-Memory errors. Requires accelerate to be installed (`pip install accelerate`)
full_determinism (`bool`, *optional*, defaults to `False`) full_determinism (`bool`, *optional*, defaults to `False`)
If `True`, [`enable_full_determinism`] is called instead of [`set_seed`] to ensure reproducible results in If `True`, [`enable_full_determinism`] is called instead of [`set_seed`] to ensure reproducible results in
distributed training distributed training. Important: this will negatively impact the performance, so only use it for debugging.
torchdynamo (`str`, *optional*): torchdynamo (`str`, *optional*):
If set, the backend compiler for TorchDynamo. Possible choices are `"eager"`, `"aot_eager"`, `"inductor"`, If set, the backend compiler for TorchDynamo. Possible choices are `"eager"`, `"aot_eager"`, `"inductor"`,
`"nvfuser"`, `"aot_nvfuser"`, `"aot_cudagraphs"`, `"ofi"`, `"fx2trt"`, `"onnxrt"` and `"ipex"`. `"nvfuser"`, `"aot_nvfuser"`, `"aot_cudagraphs"`, `"ofi"`, `"fx2trt"`, `"onnxrt"` and `"ipex"`.
...@@ -1055,7 +1055,7 @@ class TrainingArguments: ...@@ -1055,7 +1055,7 @@ class TrainingArguments:
metadata={ metadata={
"help": ( "help": (
"Whether to call enable_full_determinism instead of set_seed for reproducibility in distributed" "Whether to call enable_full_determinism instead of set_seed for reproducibility in distributed"
" training" " training. Important: this will negatively impact the performance, so only use it for debugging."
) )
}, },
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment