From be152b7616c26ef83c7bc4c3a301bb0d858c835f Mon Sep 17 00:00:00 2001
From: Jeremy Hummel <jmhummel@onica.com>
Date: Wed, 14 Aug 2019 11:52:36 -0700
Subject: [PATCH] fix typo

---
 models/Model_SAE/Model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py
index 2ddef08..f1165f1 100644
--- a/models/Model_SAE/Model.py
+++ b/models/Model_SAE/Model.py
@@ -438,7 +438,7 @@ class SAEModel(ModelBase):
             # Updated "pred_src_dst[-1]" to "pred_src_dst[-1][:4]", as it was breaking on images with 4th channel
             if self.options['learn_mask']:
                 self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4], pred_dst_dstm[-1], pred_src_dstm[-1]])
-                else:
+            else:
                 self.AE_convert = K.function([warped_dst], [pred_src_dst[-1][:4]])
 
         if self.is_training_mode: