Freeze aerial warmup across training loop

scripts/run_belgium_building_training_loop.py

@@ -105,6 +105,11 @@ def training_command(
     degrees: float = 0.0,
     flipud: float = 0.0,
     fliplr: float = 0.5,
+    warmup_epochs: float = 1.0,
+    warmup_bias_lr: float = 0.01,
+    hsv_h: float = 0.01,
+    hsv_s: float = 0.2,
+    hsv_v: float = 0.15,
 ) -> list[str]:
     command = [
         yolo,
@@ -127,6 +132,11 @@ def training_command(
         f"degrees={degrees}",
         f"flipud={flipud}",
         f"fliplr={fliplr}",
+        f"warmup_epochs={warmup_epochs}",
+        f"warmup_bias_lr={warmup_bias_lr}",
+        f"hsv_h={hsv_h}",
+        f"hsv_s={hsv_s}",
+        f"hsv_v={hsv_v}",
         f"seed={seed}",
         "deterministic=True",
         f"project={project}",
@@ -194,6 +204,11 @@ def main() -> int:
     parser.add_argument("--degrees", type=float, default=0.0)
     parser.add_argument("--flipud", type=float, default=0.0)
     parser.add_argument("--fliplr", type=float, default=0.5)
+    parser.add_argument("--warmup-epochs", type=float, default=1.0)
+    parser.add_argument("--warmup-bias-lr", type=float, default=0.01)
+    parser.add_argument("--hsv-h", type=float, default=0.01)
+    parser.add_argument("--hsv-s", type=float, default=0.2)
+    parser.add_argument("--hsv-v", type=float, default=0.15)
     parser.add_argument("--seed", type=int, default=20260731)
     parser.add_argument("--yolo", default="yolo")
     parser.add_argument("--min-aggregate-f1", type=float, default=0.55)
@@ -262,6 +277,11 @@ def main() -> int:
             degrees=args.degrees,
             flipud=args.flipud,
             fliplr=args.fliplr,
+            warmup_epochs=args.warmup_epochs,
+            warmup_bias_lr=args.warmup_bias_lr,
+            hsv_h=args.hsv_h,
+            hsv_s=args.hsv_s,
+            hsv_v=args.hsv_v,
         )
         if args.dry_run:
             print(json.dumps({"training_command": command, "evaluate_existing": evaluate_existing}, indent=2))

scripts/tower-v37-completion-command.json

@@ -53,6 +53,16 @@
   "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",
   "20260807",
   "--yolo",