Put back flips.

LCAV · Aug 22, 2024 · 2acd2ad · 2acd2ad
1 parent ed6371c
commit 2acd2ad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lensless/recon/utils.py b/lensless/recon/utils.py
@@ -873,6 +873,8 @@ def train_epoch(self, data_loader):
         for batch in pbar:
 
             # get batch
+            flip_lr = None
+            flip_ud = None
             if self.train_random_flip:
                 X, y, psfs, flip_lr, flip_ud = batch
                 psfs = psfs.to(self.device)