WassersteinGAN源码 作者的代码包括两部分:models包下包含dcgan.py和mlp.py, 这两个py文件是两种不同的网络结构,在dcgan.py中判别器和生成器都含有卷积网络,而mlp.py中判别器和生成器都只是全连接. 此外main.py为主函数,通过引入import models中的生成器和判别器来完成训练与迭代. 参数说明(main.py中): parser = argparse.ArgumentParser() parser.add_argument('--data…