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

update readme

Browse Source
This commit is contained in:
Phil Wang
2020-09-06 22:35:37 -07:00
parent e1800c1a8d
commit 1f1e42e9f9
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+2 -2
View File
@@ -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()
denoising_diffusion_pytorch/denoising_diffusion_pytorch.py
+2 -2
View File
@@ -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
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.1.0',
version = '0.1.1',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',