File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
examples/advanced_diffusion_training Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -1057,7 +1057,7 @@ def __init__(
10571057 if interpolation is None :
10581058 raise ValueError (f"Unsupported interpolation mode { interpolation = } ." )
10591059 train_resize = transforms .Resize (size , interpolation = interpolation )
1060- train_crop = transforms .CenterCrop (size ) if center_crop else transforms .RandomCrop (size )
1060+ train_crop = transforms .CenterCrop (size ) if args . center_crop else transforms .RandomCrop (size )
10611061 train_flip = transforms .RandomHorizontalFlip (p = 1.0 )
10621062 train_transforms = transforms .Compose (
10631063 [
@@ -1101,7 +1101,7 @@ def __init__(
11011101 self .image_transforms = transforms .Compose (
11021102 [
11031103 transforms .Resize (size , interpolation = interpolation ),
1104- transforms .CenterCrop (size ) if center_crop else transforms .RandomCrop (size ),
1104+ transforms .CenterCrop (size ) if args . center_crop else transforms .RandomCrop (size ),
11051105 transforms .ToTensor (),
11061106 transforms .Normalize ([0.5 ], [0.5 ]),
11071107 ]
You can’t perform that action at this time.
0 commit comments