Skip to content

Commit a25c065

Browse files
author
littletomatodonkey
authored
fix permute api to transpose (#4913)
1 parent 294ff30 commit a25c065

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dygraph/mobilenet/imagenet_dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,13 @@ def __init__(self,
3838
self.transform = transforms.Compose([
3939
transforms.RandomResizedCrop(image_size),
4040
transforms.RandomHorizontalFlip(),
41-
transforms.Permute(mode='CHW'), normalize
41+
transforms.Transpose(order=(2, 0, 1)), normalize
4242
])
4343
else:
4444
self.transform = transforms.Compose([
4545
transforms.Resize(resize_short_size),
4646
transforms.CenterCrop(image_size),
47-
transforms.Permute(mode='CHW'), normalize
47+
transforms.Transpose(order=(2, 0, 1)), normalize
4848
])
4949

5050
def __getitem__(self, idx):

0 commit comments

Comments
 (0)