fix #905

7538cc66 · Yuxin Wu · 9cdde927 · 7538cc66
Commit 7538cc66 authored Sep 21, 2018 by Yuxin Wu
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/GAN/InfoGAN-mnist.py examples/GAN/InfoGAN-mnist.py +1 -0

No files found.
--- a/examples/GAN/InfoGAN-mnist.py
+++ b/examples/GAN/InfoGAN-mnist.py
@@ -212,6 +212,7 @@ class Model(GANModelDesc):
 def get_data():
    ds = ConcatData([dataset.Mnist('train'), dataset.Mnist('test')])
    ds = BatchData(ds, BATCH)
+    ds = MapData(ds, lambda dp: [dp[0]])  # only use the image
    return ds