diff --git a/configs/deformable_detr/README.md b/configs/deformable_detr/README.md index b8a142824bfc59ca4df088bfa47d27e2fc4d109c..b8e6f953fad71e9d4527169422e2977174632a9d 100644 --- a/configs/deformable_detr/README.md +++ b/configs/deformable_detr/README.md @@ -17,6 +17,12 @@ Deformable DETR is an object detection model based on DETR. We reproduced the mo - Deformable DETR is trained on COCO train2017 dataset and evaluated on val2017 results of `mAP(IoU=0.5:0.95)`. - Deformable DETR uses 8GPU to train 50 epochs. +GPU multi-card training +```bash +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 +python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/deformable_detr/deformable_detr_r50_1x_coco.yml --fleet -o find_unused_parameters=True +``` + ## Citations ``` @inproceedings{ diff --git a/configs/detr/README.md b/configs/detr/README.md index 47b7a1198e4b8b7704cfcb13fc2cbfaf8b5e895c..3975a179796a97ae84b51ac5f73c8ad405ef69aa 100644 --- a/configs/detr/README.md +++ b/configs/detr/README.md @@ -17,6 +17,12 @@ DETR is an object detection model based on transformer. We reproduced the model - DETR is trained on COCO train2017 dataset and evaluated on val2017 results of `mAP(IoU=0.5:0.95)`. - DETR uses 8GPU to train 500 epochs. +GPU multi-card training +```bash +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 +python -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/detr/detr_r50_1x_coco.yml --fleet -o find_unused_parameters=True +``` + ## Citations ``` @inproceedings{detr,