update

README.md

@@ -8,4 +8,5 @@
 支持模型列表：
 - tianyi07/Swin-Transformer
 - qualcomm/Shufflenet-v2
-- timm/mobilenetv4_conv_small.e2400_r224_in1k
+- timm/mobilenetv4_conv_small.e2400_r224_in1k
+- nv-community/MambaVision-T-1K

enginex/MambaVision-T-1K.md

@@ -0,0 +1,14 @@
+# 运行方式
+```python
+docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
+docker run -it --shm-size=1024G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name MambaVision_pytorch   bash # 为以上拉取的docker的镜像ID替换，本镜像为：a4dd5be0ca23
+cd /path/your_code_data/MambaVision_pytorch/mamba
+pip install wheel -i https://mirrors.aliyun.com/pypi/simple/
+pip install . --no-build-isolation --no-deps
+git clone https://github.com/Dao-AILab/causal-conv1d.git
+cd causal-conv1d
+pip install e .
+cd /path/your_code_data/MambaVision_pytorch/
+pip install . --no-build-isolation --no-deps
+pip install timm==0.9.0 tensorboardX
+```