Ссылка для ultralytics/utils/autobatch.py
Примечание
Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/autobatch .py. Если ты обнаружил проблему, пожалуйста, помоги исправить ее, создав Pull Request 🛠️. Спасибо 🙏!
ultralytics.utils.autobatch.check_train_batch_size(model, imgsz=640, amp=True)
Проверь размер обучающей партии YOLO с помощью функции autobatch().
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
model |
Module
|
YOLO Модель, для которой нужно проверить размер партии. |
требуется |
imgsz |
int
|
Размер изображения, используемого для тренировки. |
640
|
amp |
bool
|
Если True, то используй автоматическую смешанную точность (AMP) для тренировки. |
True
|
Возвращается:
Тип | Описание |
---|---|
int
|
Оптимальный размер партии, вычисленный с помощью функции autobatch(). |
Исходный код в ultralytics/utils/autobatch.py
ultralytics.utils.autobatch.autobatch(model, imgsz=640, fraction=0.6, batch_size=DEFAULT_CFG.batch)
Автоматически оценивай оптимальный размер партии YOLO , чтобы использовать часть доступной памяти CUDA.
Параметры:
Имя | Тип | Описание | По умолчанию |
---|---|---|---|
model |
module
|
YOLO Модель, для которой нужно вычислить размер партии. |
требуется |
imgsz |
int
|
Размер изображения, используемый в качестве исходного для модели YOLO . По умолчанию 640. |
640
|
fraction |
float
|
Доля доступной памяти CUDA, которую нужно использовать. По умолчанию 0,60. |
0.6
|
batch_size |
int
|
Размер партии по умолчанию, который будет использоваться при обнаружении ошибки. По умолчанию - 16. |
batch
|
Возвращается:
Тип | Описание |
---|---|
int
|
Оптимальный размер партии. |