don't disable ema by default

for deterministic models at least
henryaddison · Aug 9, 2024 · 69b4742 · 69b4742
1 parent f0118a6
commit 69b4742
Showing 1 changed file with 1 addition and 0 deletions.
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
@@ -71,6 +71,7 @@ def get_default_configs():
   model.loc_spec_channels = 0
   model.num_scales = 1
   model.ema_rate = 0.9999
+  model.ema_disabled = False
   model.dropout = 0.1
   model.embedding_type = 'fourier'