feat: support PVTV2

2021-12-13 07:15:49 +00:00 · 2021-12-13 07:15:49 +00:00 · 00fb3f7519
parent a53b636829
commit 00fb3f7519
1 changed files with 1 additions and 0 deletions
--- 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]