From 26b5cab6c828741134bec6ba0f9cc895d8b74688 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 8 Sep 2020 15:03:38 -0700
Subject: [PATCH] update ema more frequently

---
 denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 2 +-
 setup.py                                                   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
index 8f0735d..4d68919 100644
--- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
@@ -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
diff --git a/setup.py b/setup.py
index 1a5fba9..d59e43d 100644
--- a/setup.py
+++ b/setup.py
@@ -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',