| device: auto | |
| seed: {value: 42, deterministic: false} | |
| dataset: | |
| type: fast | |
| output_root: ../../../result/final_preprocess/2025 | |
| catalog_path: ../../../result/final_preprocess/2025/catalog.csv | |
| input_window: | |
| past_minutes: 50 | |
| interval_minutes: 10 | |
| input_sources: [concat, hsr] | |
| labels: | |
| bt: | |
| future_minutes: 60 | |
| interval_minutes: 10 | |
| use_mask: false | |
| model: | |
| name: simvp_bt_aux | |
| params: | |
| in_shape: [6, 12, 583, 550] | |
| out_shape: [1, 1, 583, 550] | |
| bt_out_shape: [6, 1, 583, 550] | |
| hid_S: 64 | |
| hid_T: 256 | |
| N_S: 6 | |
| N_T: 8 | |
| model_type: gSTA | |
| groups: 1 | |
| mlp_ratio: 8.0 | |
| drop: 0.1 | |
| drop_path: 0.0 | |
| spatio_kernel_enc: 3 | |
| spatio_kernel_dec: 3 | |
| inference: | |
| checkpoint_path: ../../../result/training/checkpoints/model.safetensors | |
| output_dir: ../../../result/training/inference/2025 | |
| time_ranges: | |
| - {start: "202505010000", end: "202510312350"} | |
| require_label: true | |
| label_key: ci_hard | |
| batch_size: 1 | |
| num_workers: 0 | |
| output_key: ci | |
| save_legacy_batch_dim: true | |
| dtype: float32 | |
| save_summary: true | |
| skip_existing: true | |
| dataloader: {pin_memory: true} | |
| radar_mask: | |
| mode: masked | |
| source: hsr_valid_mask | |