From 2acd2ad3702f55db77d589061a54865c76187136 Mon Sep 17 00:00:00 2001
From: Eric Bezzam <ebezzam@gmail.com>
Date: Thu, 22 Aug 2024 12:20:18 +0000
Subject: [PATCH] Put back flips.

---
 lensless/recon/utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lensless/recon/utils.py b/lensless/recon/utils.py
index cc2a2787..65c6e238 100644
--- 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)