File size: 669 Bytes
aa529c9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
seed: 42
data:
  format_version: weatherbench_v1
  path: data/weatherbench.npz
  grid: [32, 64]
  samples: 24
  channels: [Z500, T850]
model:
  hidden_channels: 16
  layers: 5
  kernel_size: 5
train:
  epochs: 2
  learning_rate: 0.001
paths:
  checkpoint: result/checkpoints/weatherbench_cnn.pt
  training_metrics: result/training/metrics.json
  predictions: result/output/predictions.npz
  evaluation: result/evaluation/metrics.json
  figure: result/evaluation/comparison.png
paper_model:
  grid: [32, 64]
  input_channels: [Z500, T850]
  conv_layers: 5
  hidden_channels: 64
  kernel_size: 5
  activation: ELU
  optimizer: Adam
  loss: MSE
  test_years: [2017, 2018]