From 00121cab6ee6e47f522dd58905abd6e874fd251d Mon Sep 17 00:00:00 2001 From: Henry Addison Date: Mon, 25 Mar 2024 10:39:31 +0000 Subject: [PATCH] remove unnecessary config from debug cunet model --- .../ukcp_local_pr_1em_cunet_continuous.py | 22 ------------------- 1 file changed, 22 deletions(-) diff --git a/src/ml_downscaling_emulator/score_sde_pytorch/configs/subvpsde/ukcp_local_pr_1em_cunet_continuous.py b/src/ml_downscaling_emulator/score_sde_pytorch/configs/subvpsde/ukcp_local_pr_1em_cunet_continuous.py index b8dea1c8b..ac254d20d 100644 --- a/src/ml_downscaling_emulator/score_sde_pytorch/configs/subvpsde/ukcp_local_pr_1em_cunet_continuous.py +++ b/src/ml_downscaling_emulator/score_sde_pytorch/configs/subvpsde/ukcp_local_pr_1em_cunet_continuous.py @@ -39,27 +39,5 @@ def get_config(): # model model = config.model model.name = 'cunet' - model.scale_by_sigma = False - model.ema_rate = 0.9999 - model.normalization = 'GroupNorm' - model.nonlinearity = 'swish' - model.nf = 128 - model.ch_mult = (1, 2, 2, 2) - model.num_res_blocks = 4 - model.attn_resolutions = (16,) - model.resamp_with_conv = True - model.conditional = True - model.fir = True - model.fir_kernel = [1, 3, 3, 1] - model.skip_rescale = True - model.resblock_type = 'biggan' - model.progressive = 'none' - model.progressive_input = 'residual' - model.progressive_combine = 'sum' - model.attention_type = 'ddpm' - model.embedding_type = 'positional' - model.init_scale = 0. - model.fourier_scale = 16 - model.conv_size = 3 return config