File size: 1,153 Bytes
76d61a0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | 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
|