update

README.md

@@ -1,3 +1,11 @@
 # enginex-hygon-vc
 
-运行于【海光 DCU】系列算力卡的【视觉分类】引擎，基于 CNN 架构，支持 BEiT、MobileViT 等流行模型
+运行于【海光 DCU】系列算力卡的【视觉分类】引擎，基于 CNN 架构，支持 BEiT、MobileViT 等流行模型
+
+## 可支持模型列表
+可在项目文件夹 `/enginex` 下查看具体可支持模型文件的运行方式。
+
+支持模型列表：
+- tianyi07/Swin-Transformer
+- qualcomm/Shufflenet-v2
+- timm/mobilenetv4_conv_small.e2400_r224_in1k

enginex/Shufflenet-v2.md

@@ -0,0 +1,6 @@
+# 运行方式
+```python
+docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-py3.10-dtk24.04.3-ubuntu20.04
+
+docker run -it --name docker_name -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --network=host --shm-size=32G --privileged=true --device=/dev/kfd --ipc=host --device=/dev/dri/ --group-add video image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-py3.10-dtk24.04.3-ubuntu20.04 /bin/bash
+```

enginex/Swin-Transformer.md

@@ -0,0 +1,8 @@
+# 运行方式
+```python
+docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10
+docker run -it -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal/:/opt/hyhal/:ro --shm-size=80G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
+
+cd /your_code_path/
+pip install -r requirements.txt
+```

enginex/mobilenetv4_conv_small.e2400_r224_in1k.md

@@ -0,0 +1,8 @@
+# 运行方式
+```python
+docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-centos7.6-dtk24.04-py310
+# 为以上拉取的docker的镜像ID替换，本镜像为：c85ed27005f2
+docker run -it --shm-size=32G -v $PWD/MobileNetv4:/home/MobileNetv4 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name mobilenetv4  bash
+cd /home/MobileNetv4
+pip install -r requirements.txt # requirements.txt
+```