Unverified Commit ef1e6287 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix style (#11975)

up
parent 173e1b14
...@@ -1057,7 +1057,7 @@ class DreamBoothDataset(Dataset): ...@@ -1057,7 +1057,7 @@ class DreamBoothDataset(Dataset):
if interpolation is None: if interpolation is None:
raise ValueError(f"Unsupported interpolation mode {interpolation=}.") raise ValueError(f"Unsupported interpolation mode {interpolation=}.")
train_resize = transforms.Resize(size, interpolation=interpolation) train_resize = transforms.Resize(size, interpolation=interpolation)
train_crop = transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size) train_crop = transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size)
train_flip = transforms.RandomHorizontalFlip(p=1.0) train_flip = transforms.RandomHorizontalFlip(p=1.0)
train_transforms = transforms.Compose( train_transforms = transforms.Compose(
[ [
...@@ -1101,7 +1101,7 @@ class DreamBoothDataset(Dataset): ...@@ -1101,7 +1101,7 @@ class DreamBoothDataset(Dataset):
self.image_transforms = transforms.Compose( self.image_transforms = transforms.Compose(
[ [
transforms.Resize(size, interpolation=interpolation), transforms.Resize(size, interpolation=interpolation),
transforms.CenterCrop(size) if center_crop else transforms.RandomCrop(size), transforms.CenterCrop(size) if args.center_crop else transforms.RandomCrop(size),
transforms.ToTensor(), transforms.ToTensor(),
transforms.Normalize([0.5], [0.5]), transforms.Normalize([0.5], [0.5]),
] ]
......
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