| model: | |
| name: OneForecast | |
| input_channels: 69 | |
| output_channels: 69 | |
| # Native ERA5 721x1440 is sampled every sixth point to 121x240, then cropped to 120x240. | |
| grid_height: 120 | |
| grid_width: 240 | |
| dt_hours: 6 | |
| mesh_level: 5 | |
| processor_layers: 16 | |
| hidden_layers: 1 | |
| hidden_dim: 512 | |
| num_heads_edge: 4 | |
| num_heads_node: 4 | |
| weight_init: scratch | |
| official_checkpoint_path: /root/private_data/workspaces/yangzt01/OneForecast/best_ckpt.tar | |
| runtime: | |
| seed: 42 | |
| device: dcu | |
| output_dir: ./outputs | |
| distributed_backend: nccl | |
| training: | |
| start_epoch: 0 | |
| max_epoch: 5 | |
| learning_rate: 0.00025 | |
| weight_decay: 0.0 | |
| checkpoint_dir: ./data/checkpoint | |
| model_name: model_bak | |
| save_every_epoch: 1 | |
| max_batches: 1 | |
| finetuning: | |
| steps: 2 | |
| max_epoch: 2 | |
| learning_rate: 0.00025 | |
| max_batches: 1 | |
| model_source: trained | |
| trained_model_path: ./data/checkpoint/model_bak.tar | |
| official_checkpoint_path: /root/private_data/workspaces/yangzt01/OneForecast/best_ckpt.tar | |
| output_path: ./data/checkpoint/model_finetuned.tar | |
| datapipe: | |
| name: ERA5 | |
| task: global_weather_forecasting | |
| dataset_dir: ./data | |
| train_years: [2000] | |
| valid_years: [2001] | |
| test_years: [2002] | |
| input_steps: 1 | |
| output_steps: 1 | |
| normalize: true | |
| batch_size: 1 | |
| num_workers: 0 | |
| # Runtime distribution is enabled automatically when launched with torchrun. | |
| distributed: false | |
| # This exact order matches the official OneForecast global dataset. | |
| variables: | |
| - Z50 | |
| - Z100 | |
| - Z150 | |
| - Z200 | |
| - Z250 | |
| - Z300 | |
| - Z400 | |
| - Z500 | |
| - Z600 | |
| - Z700 | |
| - Z850 | |
| - Z925 | |
| - Z1000 | |
| - Q50 | |
| - Q100 | |
| - Q150 | |
| - Q200 | |
| - Q250 | |
| - Q300 | |
| - Q400 | |
| - Q500 | |
| - Q600 | |
| - Q700 | |
| - Q850 | |
| - Q925 | |
| - Q1000 | |
| - T50 | |
| - T100 | |
| - T150 | |
| - T200 | |
| - T250 | |
| - T300 | |
| - T400 | |
| - T500 | |
| - T600 | |
| - T700 | |
| - T850 | |
| - T925 | |
| - T1000 | |
| - U50 | |
| - U100 | |
| - U150 | |
| - U200 | |
| - U250 | |
| - U300 | |
| - U400 | |
| - U500 | |
| - U600 | |
| - U700 | |
| - U850 | |
| - U925 | |
| - U1000 | |
| - V50 | |
| - V100 | |
| - V150 | |
| - V200 | |
| - V250 | |
| - V300 | |
| - V400 | |
| - V500 | |
| - V600 | |
| - V700 | |
| - V850 | |
| - V925 | |
| - V1000 | |
| - U10M | |
| - V10M | |
| - T2M | |
| - MSLP | |
| inference: | |
| model_source: trained | |
| trained_model_path: ./data/checkpoint/model_bak.tar | |
| official_checkpoint_path: /root/private_data/workspaces/yangzt01/OneForecast/best_ckpt.tar | |
| output_dir: ./outputs/predictions | |
| max_batches: 1 | |
| visualization: | |
| input_dir: ./outputs/predictions | |
| output_dir: ./outputs/visualizations | |
| channels: [0, 18, 36, 54, 67, 68] | |