From 089f058d3e16f7af95a5d127262dd5b1690d0922 Mon Sep 17 00:00:00 2001 From: Suhwan Choi Date: Fri, 15 Aug 2025 17:26:02 +0000 Subject: [PATCH] chore: fix typo in `find_executable_batch_size` to match new 0.9 ratio --- src/transformers/trainer_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer_utils.py b/src/transformers/trainer_utils.py index fb0cc33dce2e..8f09617dc8bf 100644 --- a/src/transformers/trainer_utils.py +++ b/src/transformers/trainer_utils.py @@ -797,7 +797,7 @@ def find_executable_batch_size( """ Args: A basic decorator that will try to execute `function`. If it fails from exceptions related to out-of-memory or - CUDNN, the batch size is cut in half and passed to `function`. `function` must take in a `batch_size` parameter as + CUDNN, the batch size is multiplied by 0.9 and passed to `function`. `function` must take in a `batch_size` parameter as its first argument. function (`callable`, *optional*) A function to wrap