File size: 1,009 Bytes
1aeffbb | 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 | project:
name: glonet_reference
seed: 42
result_dir: ./result/glonet
data:
data_dir: ./data/
channels:
- temperature_0.49m
- temperature_47m
- temperature_92m
- temperature_155m
- salinity_222m
- salinity_318m
- salinity_380m
- salinity_453m
- salinity_541m
- u_643m
- u_763m
- u_902m
- u_1245m
- u_1684m
- u_2225m
- u_3220m
- u_3597m
- u_3992m
- u_4405m
- u_4833m
- u_5274m
- v_643m
- v_763m
- v_902m
- v_1245m
- v_1684m
- v_2225m
- v_3220m
- v_3597m
- v_3992m
- v_4405m
- v_4833m
- v_5274m
- ssh_surface
grid: [16, 32]
input_steps: 2
output_steps: 4
synthetic_samples: 32
batch_size: 2
model:
hidden_channels: 32
modes: [6, 8]
layers: 4
training:
epochs: 10
learning_rate: 0.0001
device: auto
checkpoint_dir: ./data/checkpoints
checkpoint: ./data/checkpoints/model_glonet.pth
pretrain_rollout_steps: 1
finetune_rollout_steps: 4
|