From a6531c82ade02840619cff0164e4b340af473019 Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Mon, 5 Aug 2024 21:48:49 +0100 Subject: [PATCH] add missing values to default config for det models NCSN++ needs things like sigma_max set --- .../configs/deterministic/default_configs.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/default_configs.py b/src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/default_configs.py index e99308be..a4025da7 100644 --- a/src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/default_configs.py +++ b/src/ml_downscaling_emulator/score_sde_pytorch/configs/deterministic/default_configs.py @@ -64,9 +64,15 @@ def get_default_configs(): # model config.model = model = ml_collections.ConfigDict() + model.sigma_min = 0.01 + model.sigma_max = 50 + model.beta_min = 0.1 + model.beta_max = 20. model.loc_spec_channels = 0 model.num_scales = 0 model.ema_rate = 0.9999 + model.dropout = 0.1 + model.embedding_type = 'fourier' # optimization config.optim = optim = ml_collections.ConfigDict()