File size: 963 Bytes
5b3329b 986404c 5b3329b 986404c 5b3329b 986404c 5b3329b | 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 44 | seed: 42
data:
path: data/corrdiff.npz
protocol: corrdiff_npz_v1
input_key: input
target_key: target
input_shape: [12, 36, 36]
target_shape: [4, 448, 448]
input_variables: [tcwv, t2m, u10m, v10m, t500, z500, u500, v500, t850, z850, u850, v850]
target_variables: [t2m, u10m, v10m, maximum_radar_reflectivity]
fake_samples: 2
model:
in_channels: 12
out_channels: 4
base_channels: 8
feature_size: 28
output_size: 448
sigma_data: 0.5
training:
regression_epochs: 1
diffusion_epochs: 1
batch_size: 1
max_batches_per_epoch: 1
learning_rate: 0.0002
num_workers: 0
amp: true
p_mean: -1.2
p_std: 1.2
sampling:
batch_size: 1
ensemble_size: 2
steps: 3
sigma_min: 0.002
sigma_max: 5.0
rho: 7.0
solver: heun
runtime:
device: auto
paths:
checkpoint: result/checkpoints/corrdiff.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation_dir: result/evaluation
|