MetNet-2 / conf /config.yaml
zhangrenchao's picture
Publish MetNet-2 reproduction
efe4fbe verified
Raw
History Blame Contribute Delete
1.17 kB
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