Files
yolov26_3d/docs/en/reference/utils/autobatch.md
2026-06-24 09:35:46 +08:00

856 B
Executable File

description, keywords
description keywords
Discover how to automatically estimate the best YOLO batch size for optimal CUDA memory usage in PyTorch using Ultralytics' autobatch utility. YOLO batch size, CUDA memory, PyTorch autobatch, Ultralytics, machine learning, optimal batch size, training batch size, YOLO model

Reference for ultralytics/utils/autobatch.py

!!! success "Improvements"

This page is sourced from [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/autobatch.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/autobatch.py). Have an improvement or example to add? Open a [Pull Request](https://docs.ultralytics.com/help/contributing/) — thank you! 🙏

::: ultralytics.utils.autobatch.check_train_batch_size





::: ultralytics.utils.autobatch.autobatch