File size: 1,165 Bytes
efe4fbe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
seed: 2026
data:
  path: data/metnet2_fake.npz
  format_version: metnet2_selected_windows_v1
  logical_shape: [641, 512, 512]
  channel_groups: {mrms_radar_history: 33, hrrr_atmosphere_history: 484, goes_satellite_history: 96, static_geography: 24, time_coordinates: 4}
  window: 32
  halo: 8
  samples: 8
model:
  input_channels: 641
  classes: 512
  width: 8
  stacks: 1
  dilations: [1, 2, 4, 8, 16, 32, 64, 128]
  lead_max_minutes: 720
training:
  batch_size: 1
  steps: 1
  learning_rate: 0.001
  class_chunk: 64
runtime:
  device: auto
  num_threads: 2
inference:
  lead_minutes: 2
  coverage: selected 32x32 window
  is_complete: false
paper_model:
  input_channels: 641
  classes: 512
  stacks: 3
  dilations: [1, 2, 4, 8, 16, 32, 64, 128]
  lead_times_minutes: {start: 2, stop: 720, step: 2}
  spatial_upscale: 4
  methods: [ConvLSTM, lead-time FiLM, dilated residual stacks, center crop, bilinear upsampling]
paths:
  checkpoint: result/checkpoints/metnet_2.pt
  training_metrics: result/training/metrics.json
  predictions: result/output/predictions.npz
  evaluation_metrics: result/evaluation/metrics.json
  comparison: result/evaluation/comparison.png