From cfa3888fbd710812cb3d68dea62dbaa96ee8905a Mon Sep 17 00:00:00 2001 From: joeloskarsson Date: Mon, 11 Nov 2024 10:37:53 +0100 Subject: [PATCH] Add eval scripts --- eval_scripts/graph_efm.sh | 15 +++++++++++++++ eval_scripts/graph_efm_ms.sh | 15 +++++++++++++++ eval_scripts/graph_fm.sh | 13 +++++++++++++ eval_scripts/graphcast.sh | 12 ++++++++++++ 4 files changed, 55 insertions(+) create mode 100755 eval_scripts/graph_efm.sh create mode 100755 eval_scripts/graph_efm_ms.sh create mode 100755 eval_scripts/graph_fm.sh create mode 100755 eval_scripts/graphcast.sh diff --git a/eval_scripts/graph_efm.sh b/eval_scripts/graph_efm.sh new file mode 100755 index 00000000..6888ac14 --- /dev/null +++ b/eval_scripts/graph_efm.sh @@ -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\ diff --git a/eval_scripts/graph_efm_ms.sh b/eval_scripts/graph_efm_ms.sh new file mode 100755 index 00000000..f4a3dcb5 --- /dev/null +++ b/eval_scripts/graph_efm_ms.sh @@ -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\ diff --git a/eval_scripts/graph_fm.sh b/eval_scripts/graph_fm.sh new file mode 100755 index 00000000..02c1e6f4 --- /dev/null +++ b/eval_scripts/graph_fm.sh @@ -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\ diff --git a/eval_scripts/graphcast.sh b/eval_scripts/graphcast.sh new file mode 100755 index 00000000..cbc2480a --- /dev/null +++ b/eval_scripts/graphcast.sh @@ -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\