上传者: 42168750 
                                    |
                                    上传时间: 2021-10-26 16:32:27
                                    |
                                    文件大小: 5.51MB
                                    |
                                    文件类型: -
                                
                            
                                
                            
                                GAN模型的Pytorch代码
这是使用相同的卷积架构的3种不同GAN模型的pytorch实现。
 DCGAN(深度卷积GAN)
 WGAN-CP(使用重量修剪的Wasserstein GAN)
 WGAN-GP(使用梯度罚分的Wasserstein GAN)
依存关系
突出的软件包是:
 麻木
scikit学习
张量流2.0
 pytorch 1.6.0
 火炬视觉0.7.0
 要快速轻松地安装所有依赖项,您应该使用pip
 pip install - r requirements . txt
训练
在Fashion-MNIST数据集上运行DCGAN模型的训练:
 python main.py --model DCGAN \
               --is_train True \
               --download True \
               -