From a25c065635959a1994236454451c368ff58014e7 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Wed, 21 Oct 2020 21:30:29 +0800
Subject: [PATCH] fix permute api to transpose (#4913)

---
 dygraph/mobilenet/imagenet_dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dygraph/mobilenet/imagenet_dataset.py b/dygraph/mobilenet/imagenet_dataset.py
index 1b39ac56..4c00fcd4 100644
--- a/dygraph/mobilenet/imagenet_dataset.py
+++ b/dygraph/mobilenet/imagenet_dataset.py
@@ -38,13 +38,13 @@ class ImageNetDataset(DatasetFolder):
             self.transform = transforms.Compose([
                 transforms.RandomResizedCrop(image_size),
                 transforms.RandomHorizontalFlip(),
-                transforms.Permute(mode='CHW'), normalize
+                transforms.Transpose(order=(2, 0, 1)), normalize
             ])
         else:
             self.transform = transforms.Compose([
                 transforms.Resize(resize_short_size),
                 transforms.CenterCrop(image_size),
-                transforms.Permute(mode='CHW'), normalize
+                transforms.Transpose(order=(2, 0, 1)), normalize
             ])
 
     def __getitem__(self, idx):
-- 
GitLab