From 1f1e42e9f933212513042866e03a23d502387342 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sun, 6 Sep 2020 22:35:37 -0700 Subject: [PATCH] update readme --- README.md | 4 ++-- denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 4 ++-- setup.py | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 0f9723f..247d685 100644 --- a/README.md +++ b/README.md @@ -61,9 +61,9 @@ trainer = Trainer( 'path/to/your/images', image_size = 128, train_batch_size = 32, - train_lr = 3e-4, + train_lr = 2e-5, train_num_steps = 100000, - gradient_accumulate_every = 1 + gradient_accumulate_every = 2 ) trainer.train() diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index f7af30f..21faaeb 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -372,9 +372,9 @@ class Trainer(object): *, image_size = 128, train_batch_size = 32, - train_lr = 3e-4, + train_lr = 2e-5, train_num_steps = 100000, - gradient_accumulate_every = 1 + gradient_accumulate_every = 2 ): super().__init__() self.model = diffusion_model diff --git a/setup.py b/setup.py index 14684ea..aa2e96d 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.1.0', + version = '0.1.1', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',