update ema more frequently

2026-09-11 12:11:43 +08:00 · 2020-09-08 15:03:38 -07:00
2 changed files with 2 additions and 2 deletions
@@ -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
@@ -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',