AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

better defaults

Browse Source
This commit is contained in:
anton-l
2022-06-15 14:36:43 +02:00
parent 8e020677ad
commit 0deeb06aac
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/train_ddpm.py
View File

@@ -12,7 +12,7 @@ from torchvision.transforms import (
Compose,
InterpolationMode,
Lambda,
RandomCrop,
CenterCrop,
RandomHorizontalFlip,
Resize,
ToTensor,
@@ -39,7 +39,7 @@ def main(args):
augmentations = Compose(
[
Resize(args.resolution, interpolation=InterpolationMode.BILINEAR),
RandomCrop(args.resolution),
CenterCrop(args.resolution),
RandomHorizontalFlip(),
ToTensor(),
Lambda(lambda x: x * 2 - 1),
@@ -136,7 +136,7 @@ if __name__ == "__main__":
parser.add_argument("--output_path", type=str, default="ddpm-model")
parser.add_argument("--batch_size", type=int, default=16)
parser.add_argument("--num_epochs", type=int, default=100)
parser.add_argument("--gradient_accumulation_steps", type=int, default=2)
parser.add_argument("--gradient_accumulation_steps", type=int, default=1)
parser.add_argument("--lr", type=float, default=1e-4)
parser.add_argument("--warmup_steps", type=int, default=500)
parser.add_argument(