Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
Salience-DETR_pytorch
Commits
c44c5466
"vscode:/vscode.git/clone" did not exist on "05abe7d635bd278c58a630e856a93dfe010fcca0"
Commit
c44c5466
authored
Jul 13, 2024
by
luopl
Browse files
Update README.md
parent
e58526fe
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
README.md
README.md
+2
-2
No files found.
README.md
View file @
c44c5466
...
@@ -47,7 +47,7 @@ pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --tr
...
@@ -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 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
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/s
ed
_pytorch
cd /path/your_code_data/s
alience_detr
_pytorch
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com
```
```
...
@@ -88,7 +88,7 @@ coco/
...
@@ -88,7 +88,7 @@ coco/
## 训练
## 训练
使用accelerate原生处理多DCU的软件包,使用HIP_VISIBLE_DEVICES来指定DCU。如果未指定,脚本将使用节点上所有可用的DCU进行训练。
使用accelerate原生处理多DCU的软件包,使用HIP_VISIBLE_DEVICES来指定DCU。如果未指定,脚本将使用节点上所有可用的DCU进行训练。
在训练前修改:configs/train_config.py文件
在训练前修改:configs/train_config.py文件
中的配置包括coco数据集地址等参数
### 单机单卡
### 单机单卡
```
```
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment