2021
03-05
03-05
Pytorch实现WGAN用于动漫头像生成
WGAN与GAN的不同去除sigmoid使用具有动量的优化方法,比如使用RMSProp要对Discriminator的权重做修整限制以确保lipschitz连续约WGAN实战卷积生成动漫头像 importtorchimporttorch.nnasnnimporttorchvision.transformsastransformsfromtorch.utils.dataimportDataLoaderfromtorchvision.utilsimportsave_imageimportosfromanime_face_generator.datasetimportImageDatasetbatch_size=32num_epoch...
继续阅读 >