HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

ultralytics 8.2.29 new fractional AutoBatch feature (#13446)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-06-09 00:28:11 +02:00 • committed by GitHub
parent 2fe0946376
commit 6a234f3639
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
12 changed files with 92 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

22
ultralytics/cfg/__init__.py
View file

@ -95,10 +95,19 @@ CLI_HELP_MSG = f"""
"""
# Define keys for arg type checks
CFG_FLOAT_KEYS = {"warmup_epochs", "box", "cls", "dfl", "degrees", "shear", "time", "workspace"}
CFG_FRACTION_KEYS = {
CFG_FLOAT_KEYS = { # integer or float arguments, i.e. x=2 and x=2.0
"warmup_epochs",
"box",
"cls",
"dfl",
"degrees",
"shear",
"time",
"workspace",
"batch",
}
CFG_FRACTION_KEYS = { # fractional float arguments with 0.0<=values<=1.0
"dropout",
"iou",
"lr0",
"lrf",
"momentum",
@ -121,11 +130,10 @@ CFG_FRACTION_KEYS = {
"conf",
"iou",
"fraction",
} # fraction floats 0.0 - 1.0
CFG_INT_KEYS = {
}
CFG_INT_KEYS = { # integer-only arguments
"epochs",
"patience",
"batch",
"workers",
"seed",
"close_mosaic",
@ -136,7 +144,7 @@ CFG_INT_KEYS = {
"nbs",
"save_period",
}
CFG_BOOL_KEYS = {
CFG_BOOL_KEYS = { # boolean-only arguments
"save",
"exist_ok",
"verbose",