WRF-ML / conf /config.yaml
zhangrenchao's picture
Publish WRF-ML reproduction
d3225f2 verified
Raw
History Blame Contribute Delete
760 Bytes
seed: 42
data:
format_version: wrf_ml_columns_v1
manifest: data/manifest.json
samples: 16
vertical_levels: 57
input_features: 10
profile_outputs: 4
boundary_outputs: 2
model:
input_features: 10
hidden_size: 16
layers: 3
profile_outputs: 4
boundary_outputs: 2
train:
epochs: 1
learning_rate: 0.001
paths:
checkpoint: result/checkpoints/wrf_ml.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation: result/evaluation/metrics.json
figure: result/evaluation/comparison.png
paper_model:
wrf_version: 4.3
horizontal_grid: [190, 170]
horizontal_resolution_km: 5
vertical_levels: 57
model_top_hpa: 10
sampling_minutes: 30
model_d_layers: 3
model_d_hidden_size: 32