plainerman
diff --git a/‎check_energy.py b/‎check_energy.py
diff --git a/‎configs/toy/dual_channel_single_gaussian.yaml
Lines changed: 2 additions & 1 deletion b/‎configs/toy/dual_channel_single_gaussian.yaml
Lines changed: 2 additions & 1 deletion
diff --git a/‎linear_interpolation.py b/‎linear_interpolation.py
@@ -11,7 +11,8 @@ num_gaussians: 1
 trainable_weights: False
 base_sigma: 2.5e-2
 
-epochs: 20000
+lr: 1e-2
+epochs: 1000
 BS: 512
 
 num_paths: 1000