wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-12 12:22:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

add exponential moving average of model, also allow passing in custom noise schedule

Browse Source
This commit is contained in:
Phil Wang
2020-09-08 14:58:09 -07:00
parent 81fb2a0386
commit 1307b3115d
3 changed files with 60 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup.py
+1 -1
View File
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
setup(
name = 'denoising-diffusion-pytorch',
packages = find_packages(),
version = '0.1.4',
version = '0.2.0',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',