Référence pour ultralytics/utils/autobatch.py
Note
Ce fichier est disponible à l'adresse https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/autobatch .py. Si tu repères un problème, aide à le corriger en contribuant à une Pull Request 🛠️. Merci 🙏 !
ultralytics.utils.autobatch.check_train_batch_size(model, imgsz=640, amp=True)
VĂ©rifie YOLO la taille du lot de formation en utilisant la fonction autobatch().
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
model |
Module
|
YOLO modèle pour vérifier la taille du lot. |
requis |
imgsz |
int
|
Taille de l'image utilisée pour la formation. |
640
|
amp |
bool
|
Si True, utilise la précision mixte automatique (AMP) pour la formation. |
True
|
Retourne :
Type | Description |
---|---|
int
|
Taille optimale du lot calculée à l'aide de la fonction autobatch(). |
Code source dans ultralytics/utils/autobatch.py
ultralytics.utils.autobatch.autobatch(model, imgsz=640, fraction=0.6, batch_size=DEFAULT_CFG.batch)
Estime automatiquement la meilleure taille de lot YOLO pour utiliser une fraction de la mémoire CUDA disponible.
Paramètres :
Nom | Type | Description | DĂ©faut |
---|---|---|---|
model |
module
|
YOLO pour calculer la taille du lot. |
requis |
imgsz |
int
|
La taille de l'image utilisée comme entrée pour le modèle YOLO . La valeur par défaut est 640. |
640
|
fraction |
float
|
La fraction de la mémoire CUDA disponible à utiliser. La valeur par défaut est 0,60. |
0.6
|
batch_size |
int
|
La taille du lot par défaut à utiliser si une erreur est détectée. La valeur par défaut est 16. |
batch
|
Retourne :
Type | Description |
---|---|
int
|
La taille optimale des lots. |
Code source dans ultralytics/utils/autobatch.py
Créé le 2023-11-12, Mis à jour le 2024-05-08
Auteurs : Burhan-Q (1), glenn-jocher (3), Laughing-q (1)