Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "954d5121ca3236fdae3702becb76cb5b78d6e163"
Commit 0deeb06a authored by anton-l's avatar anton-l
Browse files

better defaults

parent 8e020677
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/train_ddpm.py
View file @ 0deeb06a
...@@ -12,7 +12,7 @@ from torchvision.transforms import ( ...@@ -12,7 +12,7 @@ from torchvision.transforms import (
Compose, Compose,
InterpolationMode, InterpolationMode,
Lambda, Lambda,
RandomCrop, CenterCrop,
RandomHorizontalFlip, RandomHorizontalFlip,
Resize, Resize,
ToTensor, ToTensor,
...@@ -39,7 +39,7 @@ def main(args): ...@@ -39,7 +39,7 @@ def main(args):
augmentations = Compose( augmentations = Compose(
[ [
Resize(args.resolution, interpolation=InterpolationMode.BILINEAR), Resize(args.resolution, interpolation=InterpolationMode.BILINEAR),
RandomCrop(args.resolution), CenterCrop(args.resolution),
RandomHorizontalFlip(), RandomHorizontalFlip(),
ToTensor(), ToTensor(),
Lambda(lambda x: x * 2 - 1), Lambda(lambda x: x * 2 - 1),
...@@ -136,7 +136,7 @@ if __name__ == "__main__": ...@@ -136,7 +136,7 @@ if __name__ == "__main__":
parser.add_argument("--output_path", type=str, default="ddpm-model") parser.add_argument("--output_path", type=str, default="ddpm-model")
parser.add_argument("--batch_size", type=int, default=16) parser.add_argument("--batch_size", type=int, default=16)
parser.add_argument("--num_epochs", type=int, default=100) 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("--lr", type=float, default=1e-4)
parser.add_argument("--warmup_steps", type=int, default=500) parser.add_argument("--warmup_steps", type=int, default=500)
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment