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

update ema more frequently

Browse Source
This commit is contained in:
Phil Wang
2020-09-08 15:03:38 -07:00
parent 1307b3115d
commit 26b5cab6c8
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+1 -1
View File
@@ -19,7 +19,7 @@ from einops import rearrange
# constants
SAVE_AND_SAMPLE_EVERY = 1000
UPDATE_EMA_EVERY = 100
UPDATE_EMA_EVERY = 10
EXTS = ['jpg', 'png']
# helpers functions
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.2.0',
version = '0.2.1',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',