Merge branch 'master' of github.com:kaanaksit/odak

kaanaksit · Sep 12, 2023 · 4a1013d · 4a1013d
2 parents 24ce9c3 + 9f2cb07
commit 4a1013d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/odak/learn/wave/optimizers.py b/odak/learn/wave/optimizers.py
@@ -626,7 +626,7 @@ def gradient_descent(self, number_of_iterations=100, weights=[1., 1., 0., 0.]):
                 loss += weights[2] * loss_variation_hologram
                 include_pa_loss_flag = self.optimize_peak_amplitude and loss_image < self.img_loss_thres
                 if include_pa_loss_flag:
-                    loss -= self.peak_amplitude
+                    loss -= self.peak_amplitude * 1.
                 if self.method == 'conventional':
                     loss.backward()
                 else: