From bbc20c68d138a10b775805edc2723df3605aade9 Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Mon, 5 Aug 2024 20:24:05 +0100 Subject: [PATCH] switch to a 1em model for testing lotus --- bin/jasmin/lotus-wrapper | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/bin/jasmin/lotus-wrapper b/bin/jasmin/lotus-wrapper index d21588f4..17b8fca5 100755 --- a/bin/jasmin/lotus-wrapper +++ b/bin/jasmin/lotus-wrapper @@ -14,4 +14,7 @@ export WANDB_EXPERIMENT_NAME="ml-downscaling-emulator" # python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_cncsnpp.py --workdir ${DERIVED_DATA}/score-sde/deterministic/ukcp_local_pr_12em_cncsnpp/bham_pSTV --mode train -python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_plain_unet.py --workdir ${DERIVED_DATA}/score-sde/deterministic/ukcp_local_pr_12em_plain_unet/bham_pSTV --mode train +# python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_plain_unet.py --workdir ${DERIVED_DATA}/score-sde/deterministic/ukcp_local_pr_12em_plain_unet/bham_pSTV --mode train + + +python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_plain_unet.py --workdir ${DERIVED_DATA}/score-sde/deterministic/ukcp_local_pr_1em_plain_unet/bham_pSTV --mode train --config.data.dataset_name bham64_ccpm-4x_1em_psl-sphum4th-temp4th-vort4th_pr