diff --git a/configs/deformable_detr/README.md b/configs/deformable_detr/README.md index 6b24c4ffab67073df5058c1d76f82bbdf835d1b0..41ae056e3967a2f94b553838715e8f7784322c8e 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 26f83a3fd2f3dc20bd32310c9cb441da4ee278ff..cc0470e65a1d725a9ac8e1d6a94d01fdbb6893d7 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,