update readme and change device param in main.py

README.md

@@ -94,9 +94,7 @@ COCO2017
 
 ## 训练
 
-### 单机单卡/单机多卡
-
-修改train.sh中HIP_VISIBLE_DEVICES数据，实现单卡或者多卡训练
+### 单机多卡
 
 ```
 bash train.sh
@@ -129,7 +127,7 @@ bash val.sh
 
 ### 精度
 
-在COCO2017的测试集上进行单卡测试，结果如下表所示
+在COCO2017的测试集上进行单卡测试，结果如下表所示(待补充)
 
 | xxx | xxx | xxx | xxx | xxx |
 | :------: | :------: | :------: | :------: |:------: |

main.py

@@ -86,7 +86,7 @@ def get_args_parser():
 
     parser.add_argument('--output_dir', default='',
                         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')
     parser.add_argument('--seed', default=42, type=int)
     parser.add_argument('--resume', default='', help='resume from checkpoint')