From 00fb3f7519e32ab88944072570e12a963a613089 Mon Sep 17 00:00:00 2001
From: gaotingquan <gaotingquan@baidu.com>
Date: Mon, 13 Dec 2021 07:15:49 +0000
Subject: [PATCH] feat: support PVTV2

---
 ppcls/arch/backbone/model_zoo/pvt_v2.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ppcls/arch/backbone/model_zoo/pvt_v2.py b/ppcls/arch/backbone/model_zoo/pvt_v2.py
index 47271900..7c4749e7 100644
--- a/ppcls/arch/backbone/model_zoo/pvt_v2.py
+++ b/ppcls/arch/backbone/model_zoo/pvt_v2.py
@@ -45,6 +45,7 @@ MODEL_URLS = {
 __all__ = list(MODEL_URLS.keys())
 
 
+@paddle.jit.not_to_static
 def swapdim(x, dim1, dim2):
     a = list(range(len(x.shape)))
     a[dim1], a[dim2] = a[dim2], a[dim1]
-- 
GitLab