"docs/en/get_started/installation.md" did not exist on "144f1e129ed98d65e500928ab4df212149b24c5d"
Commit c44c5466 authored by luopl's avatar luopl
Browse files

Update README.md

parent e58526fe
......@@ -47,7 +47,7 @@ pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --tr
```
docker build --no-cache -t salience_detr:latest .
docker run -it --shm-size=128G -v /path/your_code_data/:/path/your_code_data/ -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name salience_detr_pytorch salience_detr bash
cd /path/your_code_data/sed_pytorch
cd /path/your_code_data/salience_detr_pytorch
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
```
......@@ -88,7 +88,7 @@ coco/
## 训练
使用accelerate原生处理多DCU的软件包,使用HIP_VISIBLE_DEVICES来指定DCU。如果未指定,脚本将使用节点上所有可用的DCU进行训练。
在训练前修改:configs/train_config.py文件
在训练前修改:configs/train_config.py文件中的配置包括coco数据集地址等参数
### 单机单卡
```
......
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