提交 1280cdaf 编写于 作者: W wangyang59

Merge branch 'gan' of https://github.com/wangyang59/book into gan

...@@ -2,6 +2,8 @@ output/ ...@@ -2,6 +2,8 @@ output/
uniform_params/ uniform_params/
cifar_params/ cifar_params/
mnist_params/ mnist_params/
*_samples/
*.png
*.log *.log
*.pyc *.pyc
data/mnist_data/ data/mnist_data/
......
...@@ -278,7 +278,7 @@ def main(): ...@@ -278,7 +278,7 @@ def main():
generator_machine.loadParameters(model_dir) generator_machine.loadParameters(model_dir)
noise = get_noise(batch_size, noise_dim) noise = get_noise(batch_size, noise_dim)
fake_samples = get_fake_samples(generator_machine, batch_size, noise) fake_samples = get_fake_samples(generator_machine, batch_size, noise)
save_results(fake_samples, "./generated_samples.png", data_source) save_results(fake_samples, "./generated_%s_samples.png" % data_source, data_source)
return return
dis_trainer = api.Trainer.create(dis_conf, dis_training_machine) dis_trainer = api.Trainer.create(dis_conf, dis_training_machine)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册