diff --git a/util/data_loader.py b/util/data_loader.py index 559b42c..5ee9bde 100644 --- a/util/data_loader.py +++ b/util/data_loader.py @@ -37,7 +37,7 @@ transform_train_largescale = transforms.Compose([ transforms.Resize(256), - transforms.RandomSizedCrop(224), + transforms.RandomResizedCrop(224), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406],