Skip to content

Commit 6ef4d82

Browse files
committed
Cherrypick PR pytorch#6642
1 parent 3894efb commit 6ef4d82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

references/classification/presets.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,9 @@ def __init__(
2020
):
2121
trans = [transforms.ToImageTensor()]
2222
trans.append(
23-
transforms.RandomResizedCrop(crop_size, interpolation=interpolation, antialias=True)
23+
transforms.CenterCrop(crop_size)
2424
if center_crop
25-
else transforms.CenterCrop(crop_size)
25+
else transforms.RandomResizedCrop(crop_size, interpolation=interpolation, antialias=True)
2626
)
2727
if hflip_prob > 0:
2828
trans.append(transforms.RandomHorizontalFlip(p=hflip_prob))

0 commit comments

Comments
 (0)