File size: 898 Bytes
9be39c5 | 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 | seed: 42
data:
root: data
format_version: convlstm_hko_synthetic_engineering_v1
train_samples: 4
test_samples: 2
input_frames: 5
output_frames: 15
channels: 1
height: 100
width: 100
interval_minutes: 6
patch_size: 2
rainfall_threshold_mm_h: 0.5
zr_a: 118.239
zr_b: 1.5241
radar_db_min: 0.0
radar_db_max: 70.0
model:
input_channels: 1
patch_size: 2
hidden_channels: [16, 16]
kernel_size: 3
output_frames: 15
paper_model:
input_channels: 1
patch_size: 2
hidden_channels: [64, 64]
kernel_size: 3
output_frames: 15
train:
epochs: 1
batch_size: 1
learning_rate: 0.001
rmsprop_alpha: 0.9
weight_decay: 0.0
gradient_clip_norm: 10.0
num_workers: 0
runtime:
device: auto
paths:
checkpoint: result/checkpoints/convlstm.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation
|