未验证 提交 a25c0656 编写于 作者: L littletomatodonkey 提交者: GitHub

fix permute api to transpose (#4913)

上级 294ff30c
...@@ -38,13 +38,13 @@ class ImageNetDataset(DatasetFolder): ...@@ -38,13 +38,13 @@ class ImageNetDataset(DatasetFolder):
self.transform = transforms.Compose([ self.transform = transforms.Compose([
transforms.RandomResizedCrop(image_size), transforms.RandomResizedCrop(image_size),
transforms.RandomHorizontalFlip(), transforms.RandomHorizontalFlip(),
transforms.Permute(mode='CHW'), normalize transforms.Transpose(order=(2, 0, 1)), normalize
]) ])
else: else:
self.transform = transforms.Compose([ self.transform = transforms.Compose([
transforms.Resize(resize_short_size), transforms.Resize(resize_short_size),
transforms.CenterCrop(image_size), transforms.CenterCrop(image_size),
transforms.Permute(mode='CHW'), normalize transforms.Transpose(order=(2, 0, 1)), normalize
]) ])
def __getitem__(self, idx): def __getitem__(self, idx):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册