| num_workers: 6 |
| train_split: 0.9 |
| seed: 3072 |
| img_size: 224 |
| embed_dim: 192 |
| history_size: 3 |
| num_preds: 1 |
| optimizer: |
| type: AdamW |
| lr: 0.0003 |
| weight_decay: 0.0001 |
| loss: |
| sigreg: |
| weight: 0.09 |
| kwargs: |
| knots: 17 |
| num_proj: 1024 |
| model: |
| _target_: jepa_shield.models.jepa.JEPA |
| encoder: |
| _target_: stable_pretraining.backbone.utils.vit_hf |
| size: tiny |
| patch_size: 14 |
| image_size: 224 |
| pretrained: false |
| use_mask_token: false |
| predictor: |
| _target_: jepa_shield.models.layers.ARPredictor |
| num_frames: 3 |
| input_dim: 192 |
| hidden_dim: 192 |
| output_dim: 192 |
| depth: 6 |
| heads: 16 |
| mlp_dim: 2048 |
| dim_head: 64 |
| dropout: 0.1 |
| emb_dropout: 0.0 |
| action_encoder: |
| _target_: jepa_shield.models.layers.Embedder |
| input_dim: 4 |
| emb_dim: 192 |
| projector: |
| _target_: jepa_shield.models.layers.MLP |
| input_dim: 192 |
| output_dim: 192 |
| hidden_dim: 2048 |
| norm_fn: |
| _target_: torch.nn.BatchNorm1d |
| _partial_: true |
| pred_proj: |
| _target_: jepa_shield.models.layers.MLP |
| input_dim: 192 |
| output_dim: 192 |
| hidden_dim: 2048 |
| norm_fn: |
| _target_: torch.nn.BatchNorm1d |
| _partial_: true |
| simulator_backend: analytic |
| simulator: |
| backend: analytic |
| source_root: third_party/s26-tk-tt-sim-dataset |
| occupancy_target_source: analytic_geometry |
| allow_dirty_hash_bound_source: false |
| task: |
| name: tractor_trailer |
| benchmark_protocol: configs/parking/benchmark/tractor_trailer_parking.yaml |
| benchmark_scenario_bank: configs/parking/benchmark/data/tractor_trailer_parking_cases.json |
| vehicle_kind: tractor_trailer |
| state_dim: 8 |
| motion_dim: 4 |
| action_dim: 2 |
| frameskip: 2 |
| image_size: 224 |
| state: |
| reference: tractor_rear_axle |
| channels: |
| - tractor_rear_x_m |
| - tractor_rear_y_m |
| - sin_theta_tractor |
| - cos_theta_tractor |
| - sin_theta_trailer |
| - cos_theta_trailer |
| - velocity_mps |
| - steering_rad |
| heading_encoding: sin_cos |
| position_unit: meter |
| heading_unit: radian |
| action: |
| channels: |
| - velocity_mps |
| - steering_rad |
| units: |
| - meter_per_second |
| - radian |
| storage: raw_physical |
| bev: |
| width_px: 224 |
| height_px: 224 |
| physical_width_m: 30.0 |
| physical_height_m: 30.0 |
| anchor: tractor_body_center |
| safety: |
| occupancy_threshold: 0.5 |
| occupancy_dilation_pixels: 0 |
| occupancy_certification_horizon: 8 |
| jackknife_limit_deg: 60.0 |
| vehicle: |
| kind: tractor_trailer |
| state_dim: 8 |
| camera_forward_offset_m: 1.141 |
| tractor_center_forward_offset_m: 1.141 |
| tractor_length_m: 5.978 |
| tractor_width_m: 2.0 |
| hitch_offset_behind_rear_axle_m: 1.15 |
| trailer_center_from_hitch_m: -3.8025 |
| trailer_length_m: 4.645 |
| trailer_width_m: 2.5 |
| data: |
| simulator_backend: analytic |
| verify_hashes: true |
| require_ready: true |
| frameskip: 2 |
| history_size: 3 |
| num_predictions: 1 |
| num_observations: 4 |
| action_dim: 2 |
| action_block_dim: 4 |
| require_action_valid: true |
| lewm: {} |
| preprocessing: |
| path: models/preprocessing.json |
| motion: |
| prediction_horizon: 8 |
| checkpoint_selection: |
| monitor: val/loss |
| mode: min |
| filename: best_val.ckpt |
| horizon_weights: uniform |
| train_continuation_chunks: |
| - 0 |
| - 1 |
| evaluation_continuation_chunks: |
| - 0 |
| run_test_after_fit: true |
| model: |
| latent_dim: 192 |
| action_dim: 4 |
| motion_dim: 4 |
| prediction_horizon: 8 |
| model_dim: 384 |
| num_layers: 3 |
| num_heads: 8 |
| feedforward_dim: 768 |
| dropout: 0.05 |
| loss: |
| huber_delta: 1.0 |
| benchmark: |
| enabled: false |
| run_during_motion_training: false |
| execution_stage: post_motion_optuna |
| require_exact_trials: 96 |
|
|
