some rolling back

april-tools · Jul 31, 2024 · 1c8de41 · 1c8de41
1 parent 1cbbd6d
commit 1c8de41
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/models.py b/src/models.py
@@ -20,7 +20,7 @@
 from cirkit.symbolic.parameters import (
     ExpParameter,
     Parameter,
-    TensorParameter,
+    TensorParameter, ScaledSigmoidParameter,
 )
 from cirkit.templates.region_graph import (
     LinearRegionGraph,
@@ -454,7 +454,7 @@ def gaussian_layer_factory(
             ),
             stddev_factory=lambda shape: Parameter.from_sequence(
                 TensorParameter(*shape, initializer=NormalInitializer(0.0, 1.0)),
-                ScaledSigmoidParameter(shape, vmin=1e-5, vmax=2.0, scale=2.0),
+                ScaledSigmoidParameter(shape, vmin=1e-5, vmax=2.0, scale=1.0),
             ),
         )
 
@@ -534,7 +534,7 @@ def gaussian_layer_factory(
             ),
             stddev_factory=lambda shape: Parameter.from_sequence(
                 TensorParameter(*shape, initializer=NormalInitializer(0.0, 1.0)),
-                ScaledSigmoidParameter(shape, vmin=1e-5, vmax=2.0, scale=2.0),
+                ScaledSigmoidParameter(shape, vmin=1e-5, vmax=2.0, scale=1.0),
             ),
         )