From e42ccd29d487e4ede62a469662086fe76b1a8789 Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Mon, 5 Aug 2024 21:43:32 +0100 Subject: [PATCH] switch to 1em NCSN++ as training wrapper --- bin/jasmin/lotus-wrapper | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bin/jasmin/lotus-wrapper b/bin/jasmin/lotus-wrapper index 8a683b9a..f576a861 100755 --- a/bin/jasmin/lotus-wrapper +++ b/bin/jasmin/lotus-wrapper @@ -18,10 +18,10 @@ 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 /gws/nopw/j04/bris_climdyn/henrya/workdirs/score-sde/deterministic/ukcp_local_pr_12em_cncsnpp/bham_pSTV --mode train # 1em NCSN++ -# python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_1em_cncsnpp.py --workdir /gws/nopw/j04/bris_climdyn/henrya/workdirs/score-sde/deterministic/ukcp_local_pr_1em_cncsnpp/bham_pSTV --mode train +python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_1em_cncsnpp.py --workdir /gws/nopw/j04/bris_climdyn/henrya/workdirs/score-sde/deterministic/ukcp_local_pr_1em_cncsnpp/bham_pSTV --mode train # 12em Plain U-Net -python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_plain_unet.py --workdir /gws/nopw/j04/bris_climdyn/henrya/workdirs/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 /gws/nopw/j04/bris_climdyn/henrya/workdirs/score-sde/deterministic/ukcp_local_pr_12em_plain_unet/bham_pSTV --mode train # 1em Plain U-Net # python bin/main.py --config src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/ukcp_local_pr_12em_plain_unet.py --workdir /gws/nopw/j04/bris_climdyn/henrya/workdirs/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