From b2091cb69862cb09ea6c462215681ecd90a577a1 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 30 Aug 2020 08:08:49 -0600
Subject: [PATCH] feamod fix

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

diff --git a/codes/models/feature_model.py b/codes/models/feature_model.py
index ef5e0918..b957612d 100644
--- a/codes/models/feature_model.py
+++ b/codes/models/feature_model.py
@@ -74,7 +74,7 @@ class FeatureModel(BaseModel):
         self.optimizer_G.zero_grad()
 
         # grey out the LR image but keep 3 channels.
-        lr = torch.mean(self.var_H, dim=1, keepdim=True)
+        lr = torch.mean(self.real_H, dim=1, keepdim=True)
         lr = lr.repeat(1, 3, 1, 1)
 
         self.fake_H = self.fea_train(lr, interpolate_factor=1)