用于生成建模的GenForce Lib
一个有效的PyTorch库,用于深度生成建模。 愿Generative Force(GenForce)与您同在。
强调
分布式培训框架。
训练速度快。
模块化设计,用于制作新模型的原型。
与相比,高度再现了StyleGAN的培训。
包含丰富的预训练GAN模型集的模型动物园,并带有。
我们还将在不久的将来支持以下功能。 请保持调整状态。
PGGAN和StyleGAN2(可能还有BigGAN)的培训。
模型训练基准。
从训练GAN编码器。
其他近期工作。
安装
通过conda创建一个虚拟环境。
conda create -n genforce python=3.7
conda activate genforce
安装torch和torchvision 。
conda install pytorch cudatoolkit=10.
评论0
最新资源