状态:存档(代码按原样提供,预计不会更新)
辉光
用于在重现结果的代码
要使用预训练的CelebA-HQ模型,请制作自己的操纵向量并运行我们的交互式演示,请检查demo文件夹。
要求
Tensorflow(已通过v1.8.0测试)
Horovod(已通过v0.13.8测试)和(Open)MPI
跑步
pip install -r requirements.txt
要设置(Open)MPI,请查看Horovod github。
下载数据集
对于小规模实验,请使用MNIST / CIFAR-10(使用keras由train.py直接下载)
对于大规模实验,使用的数据集位于Google Cloud位置https://openaipublic.azureedge.net/glow-demo/data/{dataset_name}-tfr.tar 。 数据集名称如下,我们提到了精确的预处理/
评论0
最新资源