Skip to content

Commit 77ed721

Browse files
Merge pull request #97 from nitinsethi22/patch-1
Update context_encoder.py
2 parents fcddb21 + c2d5ec4 commit 77ed721

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

context_encoder/context_encoder.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def train(self, epochs, batch_size=128, sample_interval=50):
148148
X_train = np.vstack((X_cats, X_dogs))
149149

150150
# Rescale -1 to 1
151-
X_train = X_train / 175.5 - 1.
151+
X_train = X_train / 127.5 - 1.
152152
y_train = y_train.reshape(-1, 1)
153153

154154
# Adversarial ground truths

0 commit comments

Comments
 (0)