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