File size: 1,234 Bytes
1ca0208 | 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 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 | project:
name: SEEDS_develop
model_name: SEEDS
task: generative_ensemble_emulation
seed: 42
paths:
data_dir: data
checkpoint_dir: data/checkpoint
result_dir: result
train_data: data/train.npz
val_data: data/val.npz
test_data: data/test.npz
checkpoint: data/checkpoint/model_bak.pth
data:
variables:
- mean_sea_level_pressure
- temperature_2m
- eastward_wind_850hpa
- northward_wind_850hpa
- geopotential_500hpa
- temperature_850hpa
- total_column_water_vapour
- specific_humidity_500hpa
faces: 6
height: 48
width: 48
seed_count: 2
target_count: 16
lead_time_days: 7
dtype: float32
model:
patch_size: 12
embed_dim: 768
spatial_layers: 6
field_layers: 4
sequence_layers: 6
mlp_ratio: 4
dropout: 0.0
sigma_min: 0.01
sigma_max: 100.0
diffusion_time_max: 1.0
prediction: score_noise
training:
batch_size: 1
gradient_accumulation_steps: 8
epochs: 10
learning_rate: 0.0001
weight_decay: 0.0
grad_clip_norm: 1.0
num_workers: 0
device: auto
resume: null
finetune: false
sampling:
steps: 16
data_batch_size: 1
member_batch_size: 1
solver: euler_maruyama
validation:
max_batches: null
plot_variable: temperature_2m
|