mirror of
https://github.com/ruvnet/RuView.git
synced 2026-04-28 14:09:33 +00:00
c63cf2ee77
- gcloud-train.sh: L4/A100/H100 VM provisioning, Rust build, training with --cuda, artifact download, auto-cleanup ($0.80-$8.50/hr) - training-config-sweep.json: 10 hyperparameter configs (LR, batch, backbone, windows, loss weights, warmup) - collect-training-data.py: UDP listener for 2-node ESP32 CSI recording to .csi.jsonl with interactive/batch labeling and manifest generation - benchmark-model.py: ONNX latency/throughput/PCK/FLOPs profiling with multi-model sweep comparison Co-Authored-By: claude-flow <ruv@ruv.net>
155 lines
4.3 KiB
JSON
155 lines
4.3 KiB
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
|
|
}
|
|
]
|
|
}
|