参考 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)
autobatch()関数を使用して、YOLO トレーニングバッチサイズをチェックする。
パラメーター
名称 | タイプ | 説明 | デフォルト |
---|---|---|---|
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)
利用可能なCUDAメモリのほんの一部を使用するために、最適なYOLO バッチサイズを自動的に推定します。
パラメーター
名称 | タイプ | 説明 | デフォルト |
---|---|---|---|
model |
module
|
YOLO バッチサイズを計算するモデル。 |
必須 |
imgsz |
int
|
YOLO モデルの入力として使用される画像サイズ。デフォルトは640。 |
640
|
fraction |
float
|
使用可能なCUDAメモリの割合。デフォルトは0.60。 |
0.6
|
batch_size |
int
|
エラーが検出された場合に使用するデフォルトのバッチ・サイズ。デフォルトは16。 |
batch
|
リターンズ
タイプ | 説明 |
---|---|
int
|
最適なバッチサイズ。 |