Skip to content

GitLab

Commit 5082bb1e authored by chenzk's avatar chenzk
Browse files

v1.2.3

parent bed86591
Pipeline #583 failed with stage
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
README.md
View file @ 5082bb1e
......@@ -12,6 +12,9 @@ Vision Transformer先将图像用卷积进行分块以降低计算量，再对
![img](./doc/attention.png)
## 环境配置
```
mv megatron-deepspeed-vit_pytorch megatron-deepspeed-vit # 去框架名后缀
```
### Docker（方法一）
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-23.04-py38-latest
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment