Small fixes.

LCAV · Nov 18, 2023 · 227c9ed · 227c9ed
1 parent 5ef7015
commit 227c9ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lensless/hardware/trainable_mask.py b/lensless/hardware/trainable_mask.py
@@ -40,7 +40,7 @@ def __init__(self, initial_mask, optimizer="Adam", lr=1e-3, **kwargs):
         """
         super().__init__()
         self._mask = torch.nn.Parameter(initial_mask)
-        self._optimizer = getattr(torch.optim, optimizer)([self._mask], lr=lr, **kwargs)
+        self._optimizer = getattr(torch.optim, optimizer)([self._mask], lr=lr)
         self.train_mask_vals = True
         self._counter = 0
 

diff --git a/mask_requirements.txt b/mask_requirements.txt
@@ -1,3 +1,4 @@
 sympy>=1.11.1
 perlin_numpy @ git+https://github.com/pvigier/perlin-numpy.git@5e26837db14042e51166eb6cad4c0df2c1907016
-waveprop>=0.0.8
+waveprop>=0.0.8
+slm_controller @ git+https://github.com/ebezzam/slm-controller.git