Move properties specific to score module

JonathanCrabbe · Dec 19, 2023 · ff4b9ac · ff4b9ac
1 parent f11ccea
commit ff4b9ac
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 9 deletions.
diff --git a/cmd/conf/score_model/default.yaml b/cmd/conf/score_model/default.yaml
@@ -4,10 +4,8 @@ d_model: 72
 num_layers: 10
 n_head: 12
 lr_max: 1.0e-3
-fourier_noise_scaling: ${fourier_noise_scaling}
-likelihood_weighting: ${likelihood_weighting}
-
+fourier_noise_scaling: False
+likelihood_weighting: False
 
 defaults:
   - noise_scheduler: vpsde
-
diff --git a/cmd/conf/score_model/noise_scheduler/vesde.yaml b/cmd/conf/score_model/noise_scheduler/vesde.yaml
@@ -1,5 +1,5 @@
 _target_: fdiff.schedulers.vesde_scheduler.VEScheduler
 eps: 1e-5
-sigma_min: 0.01 
+sigma_min: 0.01
 sigma_max: 2
-fourier_noise_scaling: ${fourier_noise_scaling}
+fourier_noise_scaling: ${score_model.fourier_noise_scaling}
diff --git a/cmd/conf/score_model/noise_scheduler/vpsde.yaml b/cmd/conf/score_model/noise_scheduler/vpsde.yaml
@@ -2,4 +2,4 @@ _target_: fdiff.schedulers.vpsde_scheduler.VPScheduler
 eps: 1e-5
 beta_min: 0.1
 beta_max: 20
-fourier_noise_scaling: ${fourier_noise_scaling}
+fourier_noise_scaling: ${score_model.fourier_noise_scaling}
diff --git a/cmd/conf/train.yaml b/cmd/conf/train.yaml
@@ -1,7 +1,5 @@
 random_seed: 42
 fourier_transform: True
-fourier_noise_scaling: False
-likelihood_weighting: False 
 standardize: True
 
 defaults: