Update README.md

README.md

@@ -29,7 +29,7 @@ ViT主要包括patch embeding、transformer encoder、MLP head三部分：以图
 ### Docker（方法一）
 ```
 docker pull image.sourcefind.cn:5000/dcu/admin/base/jax:0.4.23-ubuntu20.04-dtk24.04.1-py3.10
-docker run -it --network=host --privileged=true --name=docker_name --device=/dev/kfd --device=/dev/dri --group-add video --shm-size=32G  --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /path/your_code_data/:/path/your_code_data/ <imageID> /bin/bash  # <imageID>为以上拉取
+docker run -it --network=host --privileged=true --name=docker_name --device=/dev/kfd --device=/dev/dri --group-add video --shm-size=32G  --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /path/your_code_data/:/path/your_code_data/ <imageID> /bin/bash  # <imageID>为以上拉取的docker的镜像ID替换
 
 cd /your_code_path/vision_transformer
 pip install flax==0.6.9 # flax会强制安装某版本ai包
@@ -209,7 +209,7 @@ gpu推理结果：
 无
 ## 应用场景
 ### 算法类别
-`图像识别`
+`图像分类`
 ### 热点应用行业
 `制造,电商,医疗,广媒,教育`
 ## 源码仓库及问题反馈