diff --git a/README.md b/README.md index 6da93f0..550c46c 100644 --- a/README.md +++ b/README.md @@ -64,7 +64,7 @@ trainer = Trainer( diffusion, 'path/to/your/images', train_batch_size = 32, - train_lr = 2e-5, + train_lr = 1e-4, train_num_steps = 700000, # total training steps gradient_accumulate_every = 2, # gradient accumulation steps ema_decay = 0.995, # exponential moving average decay diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index fc8813b..974dc83 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -542,7 +542,7 @@ class Trainer(object): ema_decay = 0.995, image_size = 128, train_batch_size = 32, - train_lr = 2e-5, + train_lr = 1e-4, train_num_steps = 100000, gradient_accumulate_every = 2, amp = False, diff --git a/setup.py b/setup.py index 9b1b105..4e68f67 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.15.6', + version = '0.15.7', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',