Tài liệu tham khảo cho ultralytics/utils/autobatch.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/autobatch.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.utils.autobatch.check_train_batch_size(model, imgsz=640, amp=True)
Kiểm YOLO Kích thước lô đào tạo bằng hàm autobatch().
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
model |
Module
|
YOLO mô hình để kiểm tra kích thước lô cho. |
bắt buộc |
imgsz |
int
|
Kích thước hình ảnh được sử dụng để đào tạo. |
640
|
amp |
bool
|
Nếu Đúng, hãy sử dụng độ chính xác hỗn hợp tự động (AMP) để đào tạo. |
True
|
Trở lại:
Kiểu | Sự miêu tả |
---|---|
int
|
Kích thước lô tối ưu được tính bằng hàm autobatch(). |
Mã nguồn trong ultralytics/utils/autobatch.py
ultralytics.utils.autobatch.autobatch(model, imgsz=640, fraction=0.6, batch_size=DEFAULT_CFG.batch)
Tự động ước tính tốt nhất YOLO kích thước hàng loạt để sử dụng một phần của bộ nhớ CUDA có sẵn.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
model |
module
|
YOLO mô hình để tính toán kích thước lô cho. |
bắt buộc |
imgsz |
int
|
Kích thước hình ảnh được sử dụng làm đầu vào cho YOLO mẫu. Mặc định là 640. |
640
|
fraction |
float
|
Phần bộ nhớ CUDA có sẵn để sử dụng. Mặc định là 0,60. |
0.6
|
batch_size |
int
|
Kích thước lô mặc định để sử dụng nếu phát hiện lỗi. Mặc định là 16. |
batch
|
Trở lại:
Kiểu | Sự miêu tả |
---|---|
int
|
Kích thước lô tối ưu. |