Add dual channel experiment

Author: plainerman

configs/toy/dual_channel_single_gaussian.yaml

@@ -0,0 +1,14 @@
+save_dir: ./out/toy/dual_channel_single_gaussian
+
+test_system: double_well_dual_channel
+T: 1.0
+xi: 0.1
+
+num_gaussians: 1
+trainable_weights: False
+
+epochs: 20000
+BS: 512
+
+num_paths: 1000
+dt: 1e-3

configs/toy/dual_channel_two_gaussians.yaml

@@ -0,0 +1,14 @@
+save_dir: ./out/toy/dual_channel_two_gaussians
+
+test_system: double_well_dual_channel
+T: 1.0
+xi: 0.1
+
+num_gaussians: 2
+trainable_weights: False
+
+epochs: 20000
+BS: 512
+
+num_paths: 1000
+dt: 1e-3

potentials.py

@@ -40,5 +40,5 @@ def U_mueller_brown(xs, beta=1.0):
 
 double_well = (U_double_well,)
 double_well_hard = (U_double_well_hard,)
-double_well_dual_channel = (U_double_well_dual_channel,)
+double_well_dual_channel = (U_double_well_dual_channel, jnp.array([-0.5, 0]), jnp.array([0.5, 0]))
 mueller_brown = (U_mueller_brown, jnp.array([-0.55828035, 1.44169]), jnp.array([0.62361133, 0.02804632]))

systems.py

@@ -30,6 +30,8 @@ def from_name(cls, name: str) -> Self:
             U, A, B = potentials.double_well_hard
         elif name == 'double_well_dual_channel':
             U, A, B = potentials.double_well_dual_channel
+            xlim = jnp.array((-1.0, 1.0))
+            ylim = jnp.array((-1.0, 1.0))
         elif name == 'mueller_brown':
             U, A, B = potentials.mueller_brown
             xlim = jnp.array((-1.5, 0.9))