File size: 945 Bytes
9f29df6 | 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 52 53 54 55 56 57 58 59 60 | seed: 42
device: auto
data:
mode: synthetic
path: ./data/rainnet_fake.hdf5
input_steps: 4
forecast_steps: 1
interval_minutes: 5
raw_height: 900
raw_width: 900
padded_height: 928
padded_width: 928
pixel_size_km: 1
train_frames: 8
val_frames: 6
test_frames: 16
num_workers: 0
model:
in_channels: 4
out_channels: 1
dropout: 0.5
mode: regression
train:
epochs: 1
batch_size: 1
learning_rate: 0.0001
max_train_batches: 1
max_valid_batches: 1
checkpoint_last: ./weight/rainnet_last.pth
checkpoint_best: ./weight/rainnet_best.pth
paper_reference:
epochs: 10
batch_size: 2
learning_rate: 0.0001
raw_size: 900
padded_size: 928
inference:
checkpoint: ./weight/rainnet_best.pth
rollout_steps: 12
output_dir: ./result/output
evaluation:
thresholds_mm_h:
- 0.125
- 1.0
- 5.0
- 10.0
- 15.0
fss_windows_km:
- 1
- 5
- 10
- 20
result_dir: ./result
|