Bugfix

openai · Mar 9, 2019 · 441880b · 441880b
1 parent 0585b34
commit 441880b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -236,7 +236,7 @@ def train(target_vars, saver, sess, logger, dataloader, resume_iter, logdir):
                         np.random.uniform(
                             0,
                             FLAGS.rescale,
-                            FLAGS.batch_size)) > 0.05)
+                            FLAGS.batch_size) > 0.05)
                     data_corrupt[replay_mask] = replay_batch[replay_mask]
 
             feed_dict = {X_NOISE: data_corrupt, X: data, Y: label}