| { |
| "action_dim": 3, |
| "aux_dynamics_hidden_dim": null, |
| "batch_size": 1, |
| "checkpoint": "outputs/simvp_baseline/2026-05-25_08-44-43_full5ep_motionctxV2_grid2x4_fmScale3_dualWM10_delta05_MSE_dLong05_dSteer15_corr01_noDetach_conv1x1_ctrlSteerS030/checkpoints/best.pt", |
| "conditioning_dim": 32, |
| "conditioning_fusion": null, |
| "conditioning_injection": "single", |
| "conditioning_representation": "longitudinal_steer_speed", |
| "control_steer_input_scale": 0.3, |
| "device": "cuda", |
| "dual_conv1x1_fusion_stats": { |
| "bias_norm": 0.050001209756022555, |
| "enabled": true, |
| "param_count": 2080, |
| "simvp_half_weight_norm": 34.854761816360536, |
| "weight_norm": 35.060819933559266, |
| "wm_half_weight_norm": 3.7956123535264505, |
| "wm_to_simvp_weight_norm_ratio": 0.10889795700009121 |
| }, |
| "dual_fusion": "conv1x1", |
| "dual_wm_conditioning": "film", |
| "dual_wm_gated": true, |
| "dual_wm_hidden_dim": 512, |
| "dual_wm_num_layers": 3, |
| "dual_wm_scale": 1.0, |
| "future_action_detach_latents": false, |
| "future_action_future_motion_scale": 3.0, |
| "future_action_head_variant": "motion_context_v2", |
| "future_action_source": "wm", |
| "future_action_spatial_grid": [ |
| 2, |
| 4 |
| ], |
| "future_action_spatial_pooling": "grid", |
| "future_action_token_dim": 256, |
| "future_frames_per_second": 205.4795994525576, |
| "has_future_action_cls_heads": true, |
| "has_future_action_head": true, |
| "height": 320, |
| "iters": 200, |
| "latency_ms_per_batch": 38.93330540508032, |
| "model_variant": "av_wm_dual_bigwm", |
| "normalize_controls": true, |
| "parameters": 11452811, |
| "peak_vram_mb": 1243.41455078125, |
| "pred_future_actions_shape": [ |
| 1, |
| 8, |
| 2 |
| ], |
| "samples_per_second": 25.6849499315697, |
| "simvp_conditioning": "none", |
| "simvp_conditioning_stage": "input", |
| "speed_dim": 1, |
| "speed_scale": 20.0, |
| "warmup": 20, |
| "width": 512, |
| "wm_latent_residual": false, |
| "wm_residual_gated": true, |
| "wm_residual_hidden_dim": 128, |
| "wm_residual_scale": 0.1 |
| } |
|
|
