文件名称:diffusion:去噪扩散概率模型
文件大小:981KB
文件格式:ZIP
更新时间:2024-04-08 12:18:24
Python
去噪扩散概率模型 乔纳森·何(Jonathan Ho),阿杰伊·贾恩(Ajay Jain),彼得阿比尔 论文: : 网址: : 实验在Google Cloud TPU v3-8上运行。需要TensorFlow 1.15和Python 3.5,以及CPU实例的以下依赖关系(请参阅requirements.txt ): pip3 install fire pip3 install scipy pip3 install pillow pip3 install tensorflow-probability==0.8 pip3 install tensorflow-gan==0.0.0.dev0 pip3 install tensorflow-datasets==2.1.0 培训和评估脚本位于scripts/子目录中。运行训练和评估的命令在脚本顶部的注释中。数据存储在GCS存储桶中。编写脚本
【文件预览】:
diffusion-master
----.gitignore(2KB)
----diffusion_tf()
--------__init__.py(0B)
--------diffusion_utils.py(12KB)
--------models()
--------utils.py(6KB)
--------tpu_utils()
--------nn.py(4KB)
--------diffusion_utils_2.py(16KB)
----requirements.txt(2KB)
----resources()
--------samples.png(941KB)
----scripts()
--------run_cifar.py(7KB)
--------__init__.py(0B)
--------run_celebahq.py(6KB)
--------run_lsun.py(7KB)
----README.md(1KB)