File size: 767 Bytes
956cd31 | 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 | 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
|