Accelerate 提供了 find_executable_batch_size() 工具。这个工具在因 OOM(内存溢出)条件失败时重试代码,并自动降低批量大小。对于每个 OOM 条件,该算法将批量大小减半并重试代码,直到成功。