【文件属性】:
文件名称:diffusion:去噪扩散概率模型
文件大小:981KB
文件格式:ZIP
更新时间:2021-03-19 18:31:44
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)