feat: support PVTV2

00fb3f75 · gaotingquan · Tingquan Gao · a53b6368 · 00fb3f75
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

ppcls/arch/backbone/model_zoo/pvt_v2.py ppcls/arch/backbone/model_zoo/pvt_v2.py +1 -0

未找到文件。
--- 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]