ConvLSTM / conf /config.yaml
zhangrenchao's picture
Add engineering reproduction package
9be39c5 verified
Raw
History Blame Contribute Delete
898 Bytes
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