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 \
-
2021-10-26 16:32:27
5.51MB
Python
1