From f4798260d4fbc33d290c2c2144b38815793e22d3 Mon Sep 17 00:00:00 2001
From: cuicheng01 <45199522+cuicheng01@users.noreply.github.com>
Date: Mon, 5 Jul 2021 15:47:10 +0800
Subject: [PATCH] Update levit.py

---
 ppcls/arch/backbone/model_zoo/levit.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/ppcls/arch/backbone/model_zoo/levit.py b/ppcls/arch/backbone/model_zoo/levit.py
index 55c5817f..520c2b4b 100644
--- a/ppcls/arch/backbone/model_zoo/levit.py
+++ b/ppcls/arch/backbone/model_zoo/levit.py
@@ -221,8 +221,6 @@ class Subsample(nn.Layer):
 
     def forward(self, x):
         B, N, C = x.shape
-        #x = paddle.reshape(x, [B, self.resolution, self.resolution,
-        #                        C])[:, ::self.stride, ::self.stride]
         x = paddle.reshape(x, [B, self.resolution, self.resolution, C])
         end1, end2 = x.shape[1], x.shape[2]
         x = x[:, 0:end1:self.stride, 0:end2:self.stride]
-- 
GitLab