Seed failure-driven training from an evaluated incumbent
This commit is contained in:
@@ -231,6 +231,13 @@ def main() -> int:
|
||||
parser.add_argument("--workers", type=int, default=4)
|
||||
parser.add_argument("--max-det", type=int, default=1000)
|
||||
parser.add_argument("--imgsz", type=int, default=640)
|
||||
parser.add_argument(
|
||||
"--thresholds",
|
||||
type=float,
|
||||
nargs="+",
|
||||
default=[0.05, 0.075, 0.1, 0.15, 0.2, 0.25, 0.3, 0.4],
|
||||
help="Calibration-only confidence sweep used for threshold selection.",
|
||||
)
|
||||
parser.add_argument("--optimizer", default="auto")
|
||||
parser.add_argument("--lr0", type=float)
|
||||
parser.add_argument("--mosaic", type=float, default=1.0)
|
||||
@@ -368,6 +375,8 @@ def main() -> int:
|
||||
str(args.max_det),
|
||||
"--imgsz",
|
||||
str(args.imgsz),
|
||||
"--thresholds",
|
||||
*map(str, args.thresholds),
|
||||
],
|
||||
iteration_dir / f"{role}.log",
|
||||
)
|
||||
@@ -422,6 +431,7 @@ def main() -> int:
|
||||
"--device", "cuda:0",
|
||||
"--max-det", str(args.max_det),
|
||||
"--imgsz", str(args.imgsz),
|
||||
"--thresholds", *map(str, args.thresholds),
|
||||
],
|
||||
iteration_dir / f"{role}.log",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user