Change metrics settings for sampling

JonathanCrabbe · Dec 29, 2023 · 0e5eef1 · 0e5eef1
1 parent 3a0b001
commit 0e5eef1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/conf/metrics/default.yaml b/cmd/conf/metrics/default.yaml
@@ -5,7 +5,7 @@ metrics:
   - _target_: fdiff.sampling.metrics.SlicedWasserstein
     _partial_: true
     random_seed: ${random_seed}
-    num_directions: 10000
+    num_directions: 1000
     save_all_distances: true
   - _target_: fdiff.sampling.metrics.MarginalWasserstein
     _partial_: true

diff --git a/cmd/conf/sample.yaml b/cmd/conf/sample.yaml
@@ -1,6 +1,6 @@
 model_path: ${hydra:runtime.cwd}/lightning_logs
 model_id: ub0lv98f
-num_samples: 1000
+num_samples: 10000
 num_diffusion_steps: 1000
 random_seed: 42