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",
|
||||
)
|
||||
|
||||
@@ -0,0 +1,74 @@
|
||||
[
|
||||
"docker",
|
||||
"exec",
|
||||
"-d",
|
||||
"geointel",
|
||||
"/opt/geointel/venv/bin/python",
|
||||
"/app/scripts/run_belgium_building_training_loop.py",
|
||||
"--initial-model",
|
||||
"/app/storage/training/building-be-v43-v42-closed-loop-r1/iteration-005/candidate.pt",
|
||||
"--train-yaml",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/train/dataset.yaml",
|
||||
"--train-summary",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/train/yolo_tile_dataset_summary.json",
|
||||
"--dataset-audit",
|
||||
"/app/storage/training/building-be-v47-corpus-audit-r1/belgium-building-corpus-audit.json",
|
||||
"--train-quality-audit",
|
||||
"/app/storage/training/building-be-v47-yolo-quality-audit-r1/operator_yolo_dataset_quality_audit.json",
|
||||
"--calibration-summary",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/calibration/yolo_tile_dataset_summary.json",
|
||||
"--test-summary",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/test/yolo_tile_dataset_summary.json",
|
||||
"--background-summary",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/background-test/yolo_tile_dataset_summary.json",
|
||||
"--corpus-manifest",
|
||||
"/app/storage/operator-data/building-be-v47-merged-rotated-holdouts-r1/operator_samples_manifest.json",
|
||||
"--output-dir",
|
||||
"/app/storage/training/building-be-v49-v47-incumbent-loop-r1",
|
||||
"--iterations",
|
||||
"20",
|
||||
"--epochs",
|
||||
"100",
|
||||
"--patience",
|
||||
"12",
|
||||
"--batch",
|
||||
"8",
|
||||
"--workers",
|
||||
"0",
|
||||
"--max-det",
|
||||
"1000",
|
||||
"--imgsz",
|
||||
"640",
|
||||
"--optimizer",
|
||||
"AdamW",
|
||||
"--lr0",
|
||||
"0.00005",
|
||||
"--mosaic",
|
||||
"0",
|
||||
"--scale",
|
||||
"0.15",
|
||||
"--translate",
|
||||
"0.05",
|
||||
"--degrees",
|
||||
"180",
|
||||
"--flipud",
|
||||
"0.5",
|
||||
"--fliplr",
|
||||
"0.5",
|
||||
"--warmup-epochs",
|
||||
"1",
|
||||
"--warmup-bias-lr",
|
||||
"0.01",
|
||||
"--hsv-h",
|
||||
"0.01",
|
||||
"--hsv-s",
|
||||
"0.2",
|
||||
"--hsv-v",
|
||||
"0.15",
|
||||
"--seed",
|
||||
"20260949",
|
||||
"--yolo",
|
||||
"/opt/geointel/venv/bin/yolo",
|
||||
"--evaluate-initial-model"
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user