Skip to content

Commit

Permalink
Add eval scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
joeloskarsson committed Nov 11, 2024
1 parent 5ac26e6 commit cfa3888
Show file tree
Hide file tree
Showing 4 changed files with 55 additions and 0 deletions.
15 changes: 15 additions & 0 deletions eval_scripts/graph_efm.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
python train_model.py\
--dataset global_era5\
--model graph_efm\
--n_workers 16\
--n_example_pred 0\
--eval_leads 40\
--hidden_dim 256\
--prior_processor_layers 1\
--encoder_processor_layers 1\
--processor_layers 1\
--graph global_hierarchical\
--batch_size 1\
--ensemble_size 80\
--load paper_checkpoints/graph_efm.ckpt\
--eval test\
15 changes: 15 additions & 0 deletions eval_scripts/graph_efm_ms.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
python train_model.py\
--dataset global_era5\
--model graph_efm\
--n_workers 16\
--n_example_pred 0\
--eval_leads 40\
--hidden_dim 256\
--prior_processor_layers 2\
--encoder_processor_layers 4\
--processor_layers 4\
--graph global_multiscale\
--batch_size 1\
--ensemble_size 80\
--load paper_checkpoints/graph_efm_ms.ckpt\
--eval test\
13 changes: 13 additions & 0 deletions eval_scripts/graph_fm.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
python train_model.py\
--dataset global_era5\
--model graph_fm\
--n_workers 16\
--n_example_pred 0\
--eval_leads 40\
--hidden_dim 256\
--processor_layers 4\
--vertical_propnets 1\
--batch_size 1\
--graph global_hierarchical\
--load paper_checkpoints/graph_fm.ckpt\
--eval test\
12 changes: 12 additions & 0 deletions eval_scripts/graphcast.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
python train_model.py\
--dataset global_era5\
--model graphcast\
--n_workers 16\
--n_example_pred 0\
--eval_leads 40\
--hidden_dim 256\
--processor_layers 8\
--batch_size 1\
--graph global_multiscale\
--load paper_checkpoints/graphcast.ckpt\
--eval test\

0 comments on commit cfa3888

Please sign in to comment.