change of default parameters

3dem · Feb 29, 2024 · 2dd7c45 · 2dd7c45
1 parent 5760613
commit 2dd7c45
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dynamight/deformations/optimize_deformations.py b/dynamight/deformations/optimize_deformations.py
@@ -62,12 +62,12 @@ def optimize_deformations(
     mask_file: Optional[Path] = None,
     checkpoint_file: Optional[Path] = None,
     n_gaussians: int = 30000,
-    n_gaussian_widths: int = 2,
+    n_gaussian_widths: int = 1,
     n_latent_dimensions: int = 5,
     n_positional_encoding_dimensions: int = 10,
     n_linear_layers: int = 8,
     n_neurons_per_layer: int = 32,
-    n_warmup_epochs: int = 10,
+    n_warmup_epochs: int = 3,
     weight_decay: float = 0.0,
     consensus_update_rate: float = 1,
     consensus_update_decay: float = 0.95,