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

pass parameter for Trainer to EMA properly

Browse Source
This commit is contained in:
Phil Wang
2022-06-21 07:37:43 -07:00
parent 8c3609a6e3
commit 75ea49a7ef
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+2 -3
View File
@@ -590,7 +590,7 @@ class Trainer(object):
gradient_accumulate_every = 2,
amp = False,
step_start_ema = 2000,
update_ema_every = 10,
ema_update_every = 10,
save_and_sample_every = 1000,
results_folder = './results',
augment_horizontal_flip = True
@@ -599,8 +599,7 @@ class Trainer(object):
self.image_size = diffusion_model.image_size
self.model = diffusion_model
self.ema = EMA(diffusion_model, beta = ema_decay)
self.update_ema_every = update_ema_every
self.ema = EMA(diffusion_model, beta = ema_decay, update_every = ema_update_every)
self.step_start_ema = step_start_ema
self.save_and_sample_every = save_and_sample_every
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.21.0',
version = '0.21.1',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',