Update encoder.py

Fixed a bug that didn't store the options in Encoder. Suggested by @nadavbh12
LongEarW · Apr 19, 2019 · 6c8b0eb · 6c8b0eb
1 parent 6d3aa2b
commit 6c8b0eb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/models/networks/encoder.py b/models/networks/encoder.py
@@ -33,6 +33,7 @@ def __init__(self, opt):
         self.fc_var = nn.Linear(ndf * 8 * s0 * s0, 256)
 
         self.actvn = nn.LeakyReLU(0.2, False)
+        self.opt = opt
 
     def forward(self, x):
         if x.size(2) != 256 or x.size(3) != 256: