From 6cfa67d83167a0178f484f4f1e5a2a749150bddb Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Thu, 16 Jul 2020 10:18:30 -0600
Subject: [PATCH] Fix featuredisc broadcast error

---
 codes/models/SRGAN_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/codes/models/SRGAN_model.py b/codes/models/SRGAN_model.py
index 0a1a4cf5..26f1b589 100644
--- a/codes/models/SRGAN_model.py
+++ b/codes/models/SRGAN_model.py
@@ -324,8 +324,8 @@ class SRGANModel(BaseModel):
                 # Apply noise to the inputs to slow discriminator convergence.
                 var_ref = var_ref + noise
                 fake_H = fake_H + noise
-                l_d_fea_real = torch.zeros(1)
-                l_d_fea_fake = torch.zeros(1)
+                l_d_fea_real = 0
+                l_d_fea_fake = 0
                 if self.opt['train']['gan_type'] == 'pixgan_fea':
                     # Compute a feature loss which is added to the GAN loss computed later to guide the discriminator better.
                     disc_fea_scale = .5