diff --git "a/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" "b/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" index 7e0cfa23..2a0b2080 100644 --- "a/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" +++ "b/chapter6-\345\256\236\346\210\230\346\214\207\345\215\227/data/dataset.py" @@ -45,7 +45,7 @@ def __init__(self, root, transforms=None, train=True, test=False): else: self.transforms = T.Compose([ T.Resize(256), - T.RandomReSizedCrop(224), + T.RandomResizedCrop(224), T.RandomHorizontalFlip(), T.ToTensor(), normalize