| 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 | |