素食主义者
一个库,可以轻松地在PyTorch中训练各种现有的GAN(生成对抗网络)。
该库主要针对GAN用户,他们希望将现有的GAN培训技术与自己的生成器/区分器一起使用。 但是,研究人员可能还会发现GAN基类对于更快地实施新的GAN训练技术很有用。
重点是简单性并提供合理的默认值。
如何安装
您需要python 3.5或更高版本。 然后: pip install vegans
如何使用
基本思想是用户提供区分器和生成器网络,而库则负责在选定的GAN设置中训练它们:
from vegans.models.GAN import WassersteinGAN
from vegans.utils import plot_losses, plot_images
generator = ### Your generator (torch.nn.Module)
adversariat = ##
2021-09-16 20:53:20
40.43MB
Python
1