feat(codebase/cls): support vision_transformer (open-mmlab#403)

* feat(codebase/cls): support vision_transformer * style(onnx2ncnn): format cpp code, upgrade mmcls version * fix(CI): upgrade mmcv to 1.4.2 * fix(onnx2ncnn): offset out of range during fuse conv reshape * docs(vision_transformer.py): update VisionTransformer desc * docs(onnx2ncnn.cpp): add more comment * feat(onnx2ncnn.cpp): revert fuse weight * docs(onnx2ncnn.cpp): add more comment * test(vision_transformer): add test case * refactor(vision_transformer.py): use symbol rewrite layer_norm * refactor(vision_transformer): fix review * fix(attention): add missing files
humu789 · Apr 26, 2022 · 2c2d1e5 · 2c2d1e5
1 parent 01a44c0
commit 2c2d1e5
Show file tree

Hide file tree

Showing 13 changed files with 668 additions and 46 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -23,7 +23,7 @@ jobs:
       matrix:
         python-version: [3.7]
         torch: [1.8.0, 1.9.0]
-        mmcv: [1.4.0]
+        mmcv: [1.4.2]
         include:
           - torch: 1.8.0
             torch_version: torch1.8
@@ -65,7 +65,7 @@ jobs:
       matrix:
         python-version: [3.7]
         torch: [1.9.0+cu102]
-        mmcv: [1.4.0]
+        mmcv: [1.4.2]
         include:
           - torch: 1.9.0+cu102
             torch_version: torch1.9
@@ -108,7 +108,7 @@ jobs:
       matrix:
         python-version: [3.7]
         torch: [1.8.0+cu111]
-        mmcv: [1.4.0]
+        mmcv: [1.4.2]
         include:
           - torch: 1.8.0+cu111
             torch_version: torch1.8