Skip to content

Commit 015331b

Browse files
authored
Merge pull request #5 from indraastra/patch-1
Fix double-normalization bug in train_input_pipeline version of DRAW
2 parents 6c80b71 + 9bc3f50 commit 015331b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cadl/draw.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -844,7 +844,7 @@ def train_input_pipeline(
844844
utils.montage(test_xs.reshape((-1, A, B, C)), 'test_xs.png')
845845
try:
846846
while not coord.should_stop() and epoch_i < n_epochs:
847-
batch_xs = sess.run(batch) / 255.0
847+
batch_xs = sess.run(batch)
848848
noise = np.random.randn(batch_size, n_z)
849849
lx, lz = sess.run(
850850
[draw['loss_x'], draw['loss_z'], train_op],

0 commit comments

Comments
 (0)