SamudrACE / conf /config.yaml
zhangrenchao's picture
Publish SamudrACE reproduction
956cd31 verified
Raw History Blame Contribute Delete
767 Bytes
seed: 42
data:
format_version: samudrace_tiles_v1
path: data/coupled_tiles.npz
logical_grid: [180, 360]
tile_size: 8
tile_origins: [[20, 40], [80, 160], [140, 280], [50, 320]]
atmosphere_channels: 46
ocean_channels: 80
model:
hidden_dim: 16
train:
epochs: 1
learning_rate: 0.001
atmosphere_steps_per_ocean: 20
paths:
checkpoint: result/checkpoints/samudrace.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation: result/evaluation/metrics.json
figure: result/evaluation/comparison.png
paper_model:
training_years: 155
validation_years: 5
test_years: 40
atmosphere_step_hours: 6
ocean_step_days: 5
atmosphere_levels: 8
ocean_levels: 19
optimizer: AdamW
batch_size: 16