Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
大约 1 年 前同步成功

通知 115
Star 4999
Fork 1114
P
PaddleClas
提交 00fb3f75 编写于 作者: G gaotingquan 提交者： Tingquan Gao
浏览文件
操作

feat: support PVTV2

上级 a53b6368
隐藏空白更改
内联 并排
Showing with 1 addition and 0 deletion
ppcls/arch/backbone/model_zoo/pvt_v2.py
浏览文件 @ 00fb3f75
...@@ -45,6 +45,7 @@ MODEL_URLS = { ...@@ -45,6 +45,7 @@ MODEL_URLS = {
__all__ = list(MODEL_URLS.keys()) __all__ = list(MODEL_URLS.keys())
@paddle.jit.not_to_static
def swapdim(x, dim1, dim2): def swapdim(x, dim1, dim2):
a = list(range(len(x.shape))) a = list(range(len(x.shape)))
a[dim1], a[dim2] = a[dim2], a[dim1] a[dim1], a[dim2] = a[dim2], a[dim1]
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或