mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
parent
00efd64e51
commit
c5e66a40a4
@ -664,7 +664,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 cut in half and passed to `function`. `function` must take in a `batch_size` parameter as
|
||||
its first argument.
|
||||
function (`callable`, *optional*)
|
||||
A function to wrap
|
||||
|
Loading…
Reference in New Issue
Block a user