Skip to content

S
seminar-breakout
Commit 7538cc66 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix #905

parent 9cdde927
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/GAN/InfoGAN-mnist.py
View file @ 7538cc66
...@@ -212,6 +212,7 @@ class Model(GANModelDesc): ...@@ -212,6 +212,7 @@ class Model(GANModelDesc):
def get_data(): def get_data():
ds = ConcatData([dataset.Mnist('train'), dataset.Mnist('test')]) ds = ConcatData([dataset.Mnist('train'), dataset.Mnist('test')])
ds = BatchData(ds, BATCH) ds = BatchData(ds, BATCH)
ds = MapData(ds, lambda dp: [dp[0]]) # only use the image
return ds return ds
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment