From e760658fdb9956ad003013e567d444dc307004a6 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 4 Oct 2020 21:08:00 -0600
Subject: [PATCH] Another fix..

---
 codes/models/steps/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/models/steps/losses.py b/codes/models/steps/losses.py
index 1928f39f..d5caa02f 100644
--- a/codes/models/steps/losses.py
+++ b/codes/models/steps/losses.py
@@ -268,7 +268,7 @@ class GeometricSimilarityGeneratorLoss(ConfigurableLoss):
         else:
             upsampled_altered = net(*altered)
 
-        if self.gen_output_to_use:
+        if self.gen_output_to_use is not None:
             upsampled_altered = upsampled_altered[self.gen_output_to_use]
 
         # Undo alteration on HR image