remove unnecessary config from debug cunet model

henryaddison · Mar 25, 2024 · 00121ca · 00121ca
1 parent 2903575
commit 00121ca
Showing 1 changed file with 0 additions and 22 deletions.
diff --git a/...scaling_emulator/score_sde_pytorch/configs/subvpsde/ukcp_local_pr_1em_cunet_continuous.py b/...scaling_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