ruvnet--ruview/scripts/training-config-sweep.json

{
  "description": "WiFi-DensePose hyperparameter sweep — 10 configurations exploring learning rate, batch size, backbone width, window length, loss ratios, and warmup schedules.",
  "base": {
    "num_subcarriers": 56,
    "native_subcarriers": 114,
    "num_antennas_tx": 3,
    "num_antennas_rx": 3,
    "heatmap_size": 56,
    "num_keypoints": 17,
    "num_body_parts": 24,
    "weight_decay": 1e-4,
    "num_epochs": 50,
    "lr_gamma": 0.1,
    "grad_clip_norm": 1.0,
    "val_every_epochs": 1,
    "early_stopping_patience": 10,
    "save_top_k": 3,
    "use_gpu": true,
    "gpu_device_id": 0,
    "num_workers": 4,
    "seed": 42
  },
  "configs": [
    {
      "_name": "baseline",
      "_description": "Default config — reference baseline",
      "learning_rate": 1e-3,
      "batch_size": 8,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 5,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "low_lr_large_batch",
      "_description": "Lower LR with larger batch — stable convergence",
      "learning_rate": 1e-4,
      "batch_size": 16,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 10,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "high_lr_small_batch",
      "_description": "Higher LR with small batch — fast exploration",
      "learning_rate": 2e-3,
      "batch_size": 4,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 3,
      "lr_milestones": [20, 40],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "narrow_backbone",
      "_description": "128-channel backbone — faster training, lower VRAM",
      "learning_rate": 1e-3,
      "batch_size": 16,
      "backbone_channels": 128,
      "window_frames": 100,
      "warmup_epochs": 5,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "short_window",
      "_description": "50-frame window — lower latency, tests temporal sensitivity",
      "learning_rate": 5e-4,
      "batch_size": 16,
      "backbone_channels": 256,
      "window_frames": 50,
      "warmup_epochs": 5,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "keypoint_heavy",
      "_description": "Heavier keypoint loss — prioritize skeleton accuracy",
      "learning_rate": 5e-4,
      "batch_size": 8,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 5,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.5,
      "lambda_dp": 0.4,
      "lambda_tr": 0.1
    },
    {
      "_name": "contrastive_heavy",
      "_description": "Strong contrastive/transfer loss — self-supervised pretraining focus",
      "learning_rate": 5e-4,
      "batch_size": 8,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 10,
      "lr_milestones": [30, 45],
      "lambda_kp": 0.2,
      "lambda_dp": 0.3,
      "lambda_tr": 0.5
    },
    {
      "_name": "wide_backbone_long_warmup",
      "_description": "256-ch backbone + long warmup + moderate LR",
      "learning_rate": 5e-4,
      "batch_size": 8,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 10,
      "lr_milestones": [35, 48],
      "lambda_kp": 0.3,
      "lambda_dp": 0.6,
      "lambda_tr": 0.1
    },
    {
      "_name": "narrow_short_aggressive",
      "_description": "128-ch + 50-frame + high LR — fast cheap exploration",
      "learning_rate": 2e-3,
      "batch_size": 16,
      "backbone_channels": 128,
      "window_frames": 50,
      "warmup_epochs": 3,
      "lr_milestones": [20, 40],
      "lambda_kp": 0.4,
      "lambda_dp": 0.5,
      "lambda_tr": 0.1
    },
    {
      "_name": "balanced_medium",
      "_description": "Balanced loss, medium LR, medium batch — robust default",
      "learning_rate": 5e-4,
      "batch_size": 8,
      "backbone_channels": 256,
      "window_frames": 100,
      "warmup_epochs": 5,
      "lr_milestones": [25, 40],
      "lambda_kp": 0.35,
      "lambda_dp": 0.45,
      "lambda_tr": 0.2
    }
  ]
}