GLONET / conf /config.yaml
yzt15806542928's picture
Upload folder using huggingface_hub
1aeffbb verified
Raw
History Blame Contribute Delete
1.01 kB
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