diff --git a/configs/deformable_detr/README.md b/configs/deformable_detr/README.md index 41ae056e3967a2f94b553838715e8f7784322c8e..ea6840fb5c69b60e6d48708d8b732ebb56e0476a 100644 --- a/configs/deformable_detr/README.md +++ b/configs/deformable_detr/README.md @@ -20,7 +20,7 @@ Deformable DETR is an object detection model based on DETR. We reproduced the mo 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 +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 ``` ## Citations diff --git a/configs/deformable_detr/deformable_detr_r50_1x_coco.yml b/configs/deformable_detr/deformable_detr_r50_1x_coco.yml index f924c4e2e23b4a5c8b6eaad20ad340b761a476af..4ca749106d418ad7e188a3ebba33fc0cb2860279 100644 --- a/configs/deformable_detr/deformable_detr_r50_1x_coco.yml +++ b/configs/deformable_detr/deformable_detr_r50_1x_coco.yml @@ -6,3 +6,4 @@ _BASE_: [ '_base_/deformable_detr_reader.yml', ] weights: output/deformable_detr_r50_1x_coco/model_final +find_unused_parameters: True diff --git a/configs/detr/README.md b/configs/detr/README.md index cc0470e65a1d725a9ac8e1d6a94d01fdbb6893d7..2dc370f3aef79149878554dffe782f6d3f045f54 100644 --- a/configs/detr/README.md +++ b/configs/detr/README.md @@ -20,7 +20,7 @@ DETR is an object detection model based on transformer. We reproduced the model 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 +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 ``` ## Citations diff --git a/configs/detr/detr_r50_1x_coco.yml b/configs/detr/detr_r50_1x_coco.yml index 0df91bb564f0269a3e036b02fd47eabb01acec52..d8838fac01b446168338ba27fcf4c2ae1722f0ef 100644 --- a/configs/detr/detr_r50_1x_coco.yml +++ b/configs/detr/detr_r50_1x_coco.yml @@ -6,3 +6,4 @@ _BASE_: [ '_base_/detr_reader.yml', ] weights: output/detr_r50_1x_coco/model_final +find_unused_parameters: True