Models and samples for ML emulators of Met Office UK CPM

Contains model weights and samples for emulators  of the Met Office's UK CPM

• a diffusion-model-based emulator (CPMGEM, workdirs/score-sde/subvpsde/ukcp_local_pr_12em_cncsnpp_continuous/bham-4x_12em_pSTV) and a variant using less data (workdirs/score-sde/subvpsde/ukcp_local_pr_1em_cncsnpp_continuous/bham-4x_1em_pSTV_lowdata)
• a deterministic U-Net emulator (workdirs/score-sde/deterministic/ukcp_local_pr_12em_plain_unet/bham_pSTV-ema-gradcl-256-batch)

Also included are samples of bilinear interpolation of coarsned rainfall, used for comparison (workdirs/id-linpr).

For each model, samples can be found in samples/<CHECKPOINT_ID>/<DATASET>/<XFM_ID>/<SPLIT>/<ENSEMBLE_MEMBER>. e.g. in workdirs/score-sde/subvpsde/ukcp_local_pr_12em_cncsnpp_continuous/bham-4x_12em_pSTV there are folders like samples/epoch_20/bham64_gcm-4x_12em_psl-sphum4th-temp4th-vort4th_pr-historic/bham64_gcm-4x_12em_psl-sphum4th-temp4th-vort4th_pr-pixelmmsstan/test/13

For each model, checkpoints directory holds snapshots of model weights from training.

Samples and models were created using code from https://github.com/henryaddison/mlde (in particular, .