Skip to content

GitLab

Commit dc4184f4 authored by chenych's avatar chenych
Browse files

update readme and change device param in main.py

parent ba2b83b0
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
README.md
View file @ dc4184f4
...@@ -94,9 +94,7 @@ COCO2017 ...@@ -94,9 +94,7 @@ COCO2017
## 训练 ## 训练
### 单机单卡/单机多卡 ### 单机多卡
修改train.sh中HIP_VISIBLE_DEVICES数据，实现单卡或者多卡训练
``` ```
bash train.sh bash train.sh
...@@ -129,7 +127,7 @@ bash val.sh ...@@ -129,7 +127,7 @@ bash val.sh
### 精度 ### 精度
在COCO2017的测试集上进行单卡测试，结果如下表所示 在COCO2017的测试集上进行单卡测试，结果如下表所示(待补充)
| xxx | xxx | xxx | xxx | xxx | | xxx | xxx | xxx | xxx | xxx |
| :------: | :------: | :------: | :------: |:------: | | :------: | :------: | :------: | :------: |:------: |
......
main.py
View file @ dc4184f4
...@@ -86,7 +86,7 @@ def get_args_parser(): ...@@ -86,7 +86,7 @@ def get_args_parser():
parser.add_argument('--output_dir', default='', parser.add_argument('--output_dir', default='',
help='path where to save, empty for no saving') help='path where to save, empty for no saving')
parser.add_argument('--device', default='cuda:0', parser.add_argument('--device', default='cuda',
help='device to use for training / testing') help='device to use for training / testing')
parser.add_argument('--seed', default=42, type=int) parser.add_argument('--seed', default=42, type=int)
parser.add_argument('--resume', default='', help='resume from checkpoint') parser.add_argument('--resume', default='', help='resume from checkpoint')
......
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