| seed: 42 | |
| data: | |
| root: data | |
| format_version: smaat_unet_synthetic_precip_v1 | |
| train_samples: 4 | |
| test_samples: 2 | |
| input_frames: 12 | |
| output_frames: 6 | |
| height: 288 | |
| width: 288 | |
| interval_minutes: 5 | |
| precipitation_scale_mm_per_5min: 47.83 | |
| model: | |
| input_channels: 12 | |
| output_channels: 6 | |
| base_channels: 8 | |
| kernels_per_layer: 2 | |
| reduction_ratio: 4 | |
| bilinear: true | |
| paper_model: | |
| input_channels: 12 | |
| output_channels: 6 | |
| base_channels: 64 | |
| kernels_per_layer: 2 | |
| reduction_ratio: 16 | |
| bilinear: true | |
| train: | |
| epochs: 1 | |
| batch_size: 1 | |
| learning_rate: 0.001 | |
| weight_decay: 0.0 | |
| num_workers: 0 | |
| runtime: | |
| device: auto | |
| paths: | |
| checkpoint: result/checkpoints/smaat_unet.pt | |
| training_metrics: result/training/metrics.json | |
| inference_dir: result/output | |
| evaluation_dir: result/evaluation | |